python读取和保存图片的三种方法

python读取和保存图片

opencv

import cv2
#load
img = cv2.imread('0.png')
#show
cv.imshow('title', img)
#save
cv2.imwrite('1_cv2.png',img)

可以看到opencv读取的图片数据类型是uint8格式（属于numpy），uint8是专门用于存储各种图像的（包括RGB，灰度图像等），范围是从0–255。
读取的形状为（H,W,C），BGR

matplotlib

import matplotlib.pyplot as plt
# load
img = plt.imread('0.png') 
#读取后的dtype为float32（属于numpy），范围是[0,1]
#读取后的形状是（H,W,C），RGB
# show
plt.imshow(img)
plt.axis('off')
plt.show()
# save
#plt.imsave('1_plt.png',img)

还有一个保存方式叫plt.savefig(),但是使用这种方式保存的时候，保存的图片会显示空白，参考了plt.savefig保存图片一片空白或者显示内容不完整、错乱后把plt.savefig()放在了show的前面得以解决。

import matplotlib.pyplot as plt
# load
img = plt.imread('0.png') 
#读取后的shape为
# show
plt.imshow(img)
plt.axis('off')
plt.savefig('1_plt2.png')
plt.show()

PIL

from PIL import Image
# load
img = Image.open('0.png')
#读取后形状是（W,H）
# show
img.show()
#save
img.save('1_pil.png')

但是使用PIL读取的图片，matplotlib保存的图片会显示错误。
原因是PIL读取的图片格式是PIL.JpegImagePlugin.JpegImageFile，形状为(W,H)
要先把格式转化为numpy，不管是plt还是cv2.imshow,在python中只认numpy.array

总结

读取图片的方式有3种，分别是matplotlib读取图片、opencv读取图片和PIL读取图片。在读取图片后，可以统一使用matplotlib工具包显示图片。但是需要注意的是 opencv 读取图片的情况，读取的图片是BGR通道排序，需要转为RGB通道。
另外注意，matplotlib使用时，plt.imshow(image)如果没有显示图片，在其后面加上plt.show()