PIL读取图片的形状及变形

代码:
from PIL import Image

img = Image.open(“图片路径”).convert(‘RGB’)

此时图片的形状是hwc,一般来说,训练网络要的形状是whc,这里可以用numpy变形
代码如下
img=np.array(img)
img=img.transpose((1,0,2)) #hwc–>whc

猜你喜欢

转载自blog.csdn.net/qq_33345917/article/details/86676727