之前在图片预处理上就遇到过opencv和PIL,很纠结PIL读取的图片不好在opencv上使用,或者是opencv读取的图片不好在PIL上使用,以及它们如何转换为numpy类型。现在有时间,做个小实验总结一下
附上代码
import numpy as np
import cv2
from PIL import Image
import matplotlib.pyplot as plt
img_cv2 = cv2.imread("jena.jfif")
img_PIL = Image.open("jena.jfif")
array_img_PIL = np.array(img_PIL) #转为numpy的array形式
#取部分数据显示 opencv bgr排列,PIL rgb排列
print("img_cv2:\n",img_cv2[:3,0,:])
print("img_pil:\n",array_img_PIL[:3,0,:])
plt.subplot(1,2,1)
plt.title("img_PIL")
plt.imshow(array_img_PIL) #使用img_PIL也可以显示,这里直接使用array
plt.subplot(1,2,2)
plt.title("img_cv2")
plt.imshow(img_cv2)
plt.show()
#将rgb排列形式转为bgr排列形式,然后用opencv函数来显示就是正常的图片了
pil_to_cv2_img = cv2.cvtColor(array_img_PIL,cv2.COLOR_RGB2BGR)
print("img_pil_to_cv2\n",pil_to_cv2_img[:3,0,:])
cv2.imshow("pil_to_cv2_img",pil_to_cv2_img)
cv2.waitKey(0)
结果如下:<