numpy array与PIL.Image的转换

hxxjxw

已于 2022-11-01 16:06:09 修改

阅读量3k

点赞数

文章标签： python 深度学习 pytorch

于 2021-10-15 20:53:49 首次发布

本文链接：https://blog.csdn.net/hxxjxw/article/details/120790588

版权

本文介绍了如何将PIL图像转换为numpyarray，以及在使用OpenCV时需要注意的颜色空间差异。在从PILImage转为numpyarray时，形状会从(w,h)变为(h,w,3)，而在保存时，由于OpenCV默认使用BGR模式，而PIL使用RGB，因此可能需要进行色彩空间转换。转换回PILImage时，同样要注意尺寸和颜色空间的对应。在进行图像处理时，正确处理这些细节至关重要。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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的，需要转一下