问题
有些数据集的图片默认图片格式是[chanel][height][width], 然而有些地方显示图片是[height][width][chanel], 这就需要改变一下图片轴的次序。
numpy.rollaxis()方法
old_img是3*32*32的,new_img是32*32*3的
newimg = np.rollaxis(old_img, 0, 3) # 把轴0放到轴3的位置
其他
如果只是为了显示图片的话,scipy.misc.toimage()方法很好用
from scipy.misc import toimage
plt.imshow(toimage(old_img))
plt.show()