最近调试程序的时候,发现要保存的图片颜色与原图片有色差(显示图像变蓝),郁闷良久,终于发现问题,望诸君避坑。
在用plt.imshow和cv2.imshow显示同一幅图时可能会出现颜色差别很大的现象。
原因:opencv的接口使用BGR,而matplotlib.pyplot 则是RGB模式。
单独使用图像是正常的,联合使用显示是不正常的。
下面我们根据程序结合结果来看:
import cv2 as cv
import numpy as np
import matplotlib.pyplot as plt
img = cv.imread('road.jpg') # CV 读取图片
img2 = plt.imread('road.jpg') # plt 读取图片
img3 = cv.cvtColor(img, cv.COLOR_BGR2RGB) # CV BGR转变RGB
img4 = cv.cvtColor(img2, cv.COLOR_BGR2RGB) # Plt BGR转变RGB
# Plt
plt.imshow(img)
plt.savefig('cv_read_plt_show.jpg')
plt.show()
plt.imshow(img2)
plt.savefig('plt_read_plt_show.jpg')
plt.show()
plt.imshow(img3)
plt.savefig('cv_read_plt_show_BGR2RGB.jpg')
plt.show()
plt.imshow(img4)
plt.savefig('plt_read_plt_show_BGR2RGB.jpg')
plt.show()
# CV
cv.namedWindow("bgr_image", 0)
cv.resizeWindow("bgr_image", 512, 768)
cv.imshow('bgr_image', img)
cv.imwrite('cv_read_cv_show.jpg',img)
cv.namedWindow("rgb_image", 0);
cv.resizeWindow("rgb_image", 512, 768)
cv.imshow('rgb_image', img2)
cv.imwrite('plt_read_cv_show.jpg',img2)
cv.namedWindow("cv_read_BGR2RGB_image", 0);
cv.resizeWindow("cv_read_BGR2RGB_image", 512, 768)
cv.imshow('cv_read_BGR2RGB_image', img3)
cv.imwrite('cv_read_cv_show_BGR2RGB.jpg', img3)
cv.namedWindow("plt_read_BGR2RGB_image", 0);
cv.resizeWindow("plt_read_BGR2RGB_image", 512, 768)
cv.imshow('plt_read_BGR2RGB_image', img4)
cv.imwrite('plt_read_cv_show_BGR2RGB.jpg', img4)
# cv.waitKey(0)
# cv.destroyAllWindows()