PIL和opencv

from PIL import Image, ImageEnhance
img = Image.open(img_path)
img_np=np.array(img).astype('float32')
img_np.shape = (HWC)
img_np.transpose((2, 0, 1)
img_np.shape = (CHW)

PIL读取进来一般是RGB，如果不是可以转换。

if img.mode != 'RGB':
	img = img.convert('RGB')

img = cv2.imread(img_path)
img_np=np.array(img).astype('float32')
img_np.shape = (HWC)
img_np.transpose((2, 0, 1)
img_np.shape = (CHW)

对于图片减均值

img.shape = (HWC)
mean = (C)
img -= mean

img.shape = (CHW)
mean = np.array([0.485, 0.456, 0.406]).reshape((3, 1, 1))
// mean.shape=(c,1,1)
img -= mean