代码:
from PIL import Image
img = Image.open(“图片路径”).convert(‘RGB’)
此时图片的形状是hwc,一般来说,训练网络要的形状是whc,这里可以用numpy变形
代码如下
img=np.array(img)
img=img.transpose((1,0,2)) #hwc–>whc
代码:
from PIL import Image
img = Image.open(“图片路径”).convert(‘RGB’)
此时图片的形状是hwc,一般来说,训练网络要的形状是whc,这里可以用numpy变形
代码如下
img=np.array(img)
img=img.transpose((1,0,2)) #hwc–>whc