深度学习-图片

1 pytorch 对图片增加维度

要求图片4维： 数量* 通道数* 长*宽，二本地选取单张图片只有:通道数* 长* 宽用numpy.reshape() 可以对其扩展。

import numpy 

from PIL import Image

img=Image.open(“/home/1,jpg”)

img=np.array(img）

img=img.reshape(1,3,32,32)

这样就不会因为维度的问题再报错，神经网络要求tensor 类型，注意转换。