官网https://docs.opencv.org/3.4.1/dd/d0d/tutorial_py_2d_histogram.html
前面的文章中介绍了如何计算和绘制一维直方图。之所以称为一维,因为我们只关注图像的1个特征,例如:像素的灰度值。在二维直方图中,我们需要考虑2个特征,通常它被用于颜色直方图。颜色直方图需要考虑2个特征,像素的颜色(Hue)和饱和度(Saturation)。
opencv官方例子samples/python/color_histogram.py中有颜色直方图绘制,本节会对其介绍。这对于后面直方图投影学习有帮助。
1.opencv中的2D直方图
opencv提供cv.calcHist()函数计算直方图。
对于彩色直方图,我们需要把颜色空间从BGR转换到HSV(如果是1D直方图,需要从BGR转换到灰度图)。
2D直方图,参数需要做如下修改:
- channels = [0,1] 因为我们需要同时处理H和S通道
- bins = [180,256] 180-H通道,256-S通道
- range = [0,180,0,256] H范围0至180,S范围0至256
代码例子如下
import numpy as np
import cv2 as cv
img = cv.imread('result.jpg')
hsv = cv.cvtColor(img,cv.COLOR_BGR2HSV)
hist = cv.calcHist([hsv], [0, 1], None, [180, 256], [0, 180, 0, 256])