用numpy.rollaxis改变图片轴次序

问题

有些数据集的图片默认图片格式是[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()