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 类型,注意转换。
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 类型,注意转换。