python中读取图片

from PIL import Image
img = Image.open(image_path)
np_img = np.array(img)
#np_img = np.delete(np_img, -1, axis = 2)
np.img.transpose((2,0,1))
img.show()可以显示图片

上面读取出来的是PIL图片类型

如果要使用ndarray类型，那么就需要np.array(img)，将PIL图片类型转化成ndarray

同时如果要HWC换到CWH，那么就需要ndarray.transpose(2,0,1)

有时候读出来的图像可能会含有alpha通道，那么就需要delete一下

其实，toTensor可以将PIL图片或者unit8的ndarray数据转换成CWH的位于（0.0，1.0）的float32的tensor