可使用OpenCV的cv2.calcHist()函数查找直方图,再利用matplotlib.pyplot的plot()函数绘制直方图。
cv2.calcHist()函数的基本格式如下。
hist=cv2.calcHist(image,channels,mask,histSize,ranges)
参数说明如下。
hist为返回的直方图,是一个一维数组,其大小为256,保存了原图像中各个灰度级的数量。
image为原图像,实际参数需用方括号括起来。
channels为通道编号。灰度图像的通道编号为[0],BGR图像的通道编号为[0][1][2]。
mask为掩模图像,为None时统计整个图像,否则统计部分图像。
histSize为BINS的值,实际参数需用方括号括起来,如[256]。
ranges为像素值范围,8位灰度图像为[0,255]。
示例代码如下。
test6-2.py:查找和绘制直方图
import cv2
import matplotlib.pyplot as plt
img=cv2.imread('gate.jpg') #读取图像
cv2.imshow('original',img) #显示原图像
histb=cv2.calcHist([img],[0],None,[256],[0,255]) #计算B通道直方图
histg=cv2.calcHist([img],[1],None,[256],[0,255]) #计算G通道直方图
histr=cv2.calcHist([img],[2],None,[256],[0,255]) #计算R通道直方图
plt.plot(histb,color='b') #绘制B通道直方图,蓝色
plt.plot(histg,color='g') #绘制G通道直方图,绿色
plt.plot(histr,color='r') #绘制R通道直方图,红色
plt.show() #显示直方图