numpy array与PIL.Image的转换

PIL Image转numpy array

ndarray = np.asarray(PIL.Image.open(path))

但是注意用PIL.Image.open(path)得到的是(w,h)

而用np.asarray(PIL.Image.open(path))转成numpy array就是(h,w,3)的shape了

转成numpy array之后大概率是用cv2保存,但是注意cv2是BGR的,而PIL Image是RBG的,需要转一下

img_new = cv2.cvtColor(img_new, cv2.COLOR_RGB2BGR)

numpy array转PIL Image

image = PIL.Image.fromarray(ndarray.astype(np.uint8))
img = cv2.imread("1.jpg")  
img = cv2.cvtColor(np.asarray(im), cv2.COLOR_RGB2BGR)
image = Image.fromarray(img)

这里也是,cv2读的img是(h,w,3),转成PIL Image后size是(w,h)

注意cv2是BGR的,而PIL Image是RBG的,需要转一下

你可能感兴趣的:(python,深度学习,pytorch)