matplotlib显示彩色图像的问题,plt.imshow

问题：

在使用plt.imshow显示彩色图像的时候，出现了这样的错误：

Invalid shape (3, 512, 512) for image data

解决：

是因为一般神经网络中对图像处理之后的格式是（3，512，512）这种，分别为通道，高，宽。但是plt显示的图像格式为（512，512，3）也就是高，宽，通道。所以会出现错误。

使用下面的代码将格式转变过来就可以了：

    img = np.asarray(img)
    plt.imshow(img.transpose((1, 2, 0)))