import cv2
import numpy as np
def ImageHist(image,type): #直方图统计函数
color = (255,255,255)
windowName = 'Gray'
if type == 31:
color = (255,0,0)
windowName = 'B Hist'
elif type == 32:
color = (0,255,0)
windowName = 'G Hist'
elif type == 33:
color = (0,0,255)
windowName = 'R Hist'
# 1 image 2 [0] 3 mask None 4 256 5 0-255
#统计直方图,此处使用灰度直方图,使用第一通道,所以是[0],
#第三个参数:mask的模板,此处无,使用None;
#第四个参数:直方图柱状的个数;
#第五个参数:表面直方图张各个像素的值 (0-255,表面所有的像素都遍历一次)
hist = cv2.calcHist([image],[0],None,[256],[0.0,255.0])
minV,maxV,minL,maxL = cv2.minMaxLoc(hist) #最小值,最大值,最小值下标,最大值下标,用于下面的归一化
histImg = np.zeros([256,256,3],np.uint8) #创建一个画布 3表面rgb三个颜色
for h in range(256): #使用循环一次在画布上画图,h表示高度
intenNormal = int(hist[h]*256/maxV) #因为数值比较大,需要归一化处理
cv2.line(histImg,(h,256),(h,256-intenNormal),color)
cv2.imshow(windowName,histImg)
return histImg
img = cv2.imread('image0.jpg',1)
channels = cv2.split(img)# RGB - R G B
for i in range(0,3):
ImageHist(channels[i],31+i)
cv2.waitKey(0)
网友评论