python-从具有3个通道的numpy数组中绘制彩色图像

在我的Jupyter笔记本中,我试图显示通过Keras迭代的图像.我正在使用的代码如下

def plotImages(path, num):
 batchGenerator = file_utils.fileBatchGenerator(path+"train/", num)
 imgs,labels = next(batchGenerator)
 fig = plt.figure(figsize=(224, 224))
 plt.gray()
 for i in range(num):
    sub = fig.add_subplot(num, 1, i + 1)
    sub.imshow(imgs[i,0], interpolation='nearest')

但这仅绘制单个通道,因此我的图像是灰度的.如何使用3个通道输出彩色图像图. ?

最佳答案

如果要显示RGB图像,则必须提供所有三个通道.根据您的代码,您只显示第一个通道,因此matplotlib没有信息将其显示为RGB.相反,由于调用了plt.gray(),它将把值映射到灰色颜色图

相反,您需要将RGB图像的所有通道传递给imshow,然后使用真彩色显示,而忽略图形的颜色图

sub.imshow(imgs, interpolation='nearest')

更新资料

由于img实际上是2 x 3 x 224 x 224,因此在显示图像之前,您需要将img索引并排列尺寸为224 x 224 x 3

im2display = imgs[1].transpose((1,2,0))
sub.imshow(im2display, interpolation='nearest')