qq_40877238的博客

原创 pytorch网络输入图片通道在前在后（channel_first和channel_last）的问题

刚开始学习pytorch卷积神经网络的时候，网络输入要求是batch33232,我们如果想要测试自己电脑上的图片格式为3232*3。即网络要求channel_first,本地图片是channel_last,此时我们只需要使用numpy中的np.transpose()函数调整下通道的顺序即可。代码如下：import numpy as npimport cv2path = r"C:\Users\X_man\Desktop\image\cat.jpg"image = cv2.imread(path,0)