plt.imshow()显示图片色差问题


转载:https://www.cnblogs.com/darkknightzh/p/6039667.html

由于系统缺少某些库,导致cv2.imshow()无法使用,于是使用matplotlib.pyplot.imshow()代替,但是在使用时发现plt.imshow()显示的图片颜色有问题。

原图:

from matplotlib import pyplot as plt import cv2 import numpy as np img = cv2.imread('XXX') plt.imshow(img)
  • 1
  • 2
  • 3
  • 4
  • 5


明显看出颜色有问题。网上查找发现原来是cv2读取的问题:
使用cv2.imread()接口读图像,读进来的是BGR格式以及【0~255】。所以只要将img转换为RGB格式显示即可:

img_2 = img[:,:,[2,1,0]] plt.imshow(img_2)
  • 1
  • 2


也正是由于这个原因,在使用pycaffe做识别时需要转换通道:
image = caffe.io.load_image(image_file) #加载图片
使用caffe.io.load_image()读取进来的是RGB格式和【0~1】(float),所以在进行识别之前要在transformer中设置transformer.set_raw_scale(‘data’,255)(缩放至0~255),以及transformer.set_channel_swap(‘data’,(2,1,0))(将RGB变换到BGR,因为训练时用cv2读取的图片是BGR的,测试需要和训练一致)


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM