Python对图片增加维度

       学习pytorch框架时发现的一个技巧，选中本地图片经过神经网络提取特征时，由于神经网络要求图片四维——数量×通道数×长×宽，而本地选取单张图片只有：通道数×长×宽，用numpy.reshape()可以对其扩维。

import numpy 
from PIL import Image

img = Image.open('/home/1.jpg') 
#将PIL转换为numpy.array才能用reshape函数  
img = np.array(img)    
img = img.reshape(1,3,32,32)

这样就不会因为维度问题再报错。

PS：神经网络要求tensor类型，注意转换

 

来源：CSDN

作者：心里有党

链接：https://blog.csdn.net/weixin_39386156/article/details/89388776