1. 程式人生 > >用numpy.rollaxis改變圖片軸次序

用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
()