我正在尝试从一个lmdb数据集
中读取图像,对每个图像进行增强,然后将它们保存到另一个数据集
中以供我的训练使用。
当这些图像被保存到lmdb数据集
中时,它们的轴最初被更改为(3,32,32)
,因此为了增强它们,我必须将它们转置回其实际形状。
问题是,每当我尝试使用matplotlib
的show()
方法或scipy
的toimage()
方法显示它们时,它们会显示图像的旋转版本。
因此我们有:
img_set = np.transpose(data_train,(0,3,2,1))
#trying to display an image using pyplot, makes it look like this:
plt.subplot(1,2,1)
plt.imshow(img_set[0])
使用 toimage
显示同一张图片:
现在,如果我不对 data_train
进行转置,pyplot
的 show()
会生成一个错误,而 toimage()
则正常显示图片:
这是怎么回事呢?
当我将转置后的数据集 data_train 输入至增强器时,得到的结果与之前的例子一样旋转了。
现在我不确定这是一个显示问题,还是实际上图像确实是旋转了!
我该怎么办?