不应该直接修改 _train_samples,因为要在 TF 中使用 4 维的 Tensor。
应该在 Inspect 中判断 shape[3] 是否为 1。
imshow 接受的 shape 有 (x, y), (x, y, 3) and (x, y, 4)
如果shape[3] == 1, 那么 reshape

def inspect(dataset, labels, i):

# 显示图片看看
if dataset.shape[3] == 1:
 shape = dataset.shape
 dataset = dataset.reshape(shape[0], shape[1], shape[2])
print(labels[i])
plt.imshow(dataset[i])
plt.show()

你好,为啥按照这个代码显示的灰度图还是有颜色啊?

@SMZCC 这是plt函数实现造成的。不是bug。不用在意。

• 蟹蟹,我知道了;
• 另外,我发现,在plt.imshow(samples[i])中,添加第二个参数,plt.imshow(samples[i], plt.cm.gray)就会灰度显示了;
• 可是,我还有个问题:我调用了inspect函数两次,理论来说应该会显示两个图片,然而,每次我都需要将第一张图片(3通道的图片)关闭掉才会显示第二张图片(单通道的),怎么让他们同时显示呢?

@SMZCC 这个你可以google一下。matplotlib不是很易用。我之前遇到过同样的问题,但是忘记了。