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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章