python Matplotlib 显示图片

若我们想使用Matplotlib 将数据显示成正常图片的形式，则使用的数据首先必须是数组，其次这个数组的维度必须是(height,weight,channel),即色彩通道数在最后面。

所以需要通过numpy 和 transpose 完成原始图像数据类型的转换和数据维度的交换

示例代码：

#假设原图是（batchsize，channel,height,weight）
#下面的方法，将这些图合并放在一个方格里面
img = torchvision.utils.make_grid(images)
#最后转换成(height,weight,channel)
img = img.numpy().transpose(1,2,0)

plt.imshow(img)