当前位置: 首页 > 文档资料 > OpenCV 中文文档 >

4.9.3 二维直方图

优质
小牛编辑
124浏览
2023-12-01

目标

在本章中,我们将学习查找和绘制二维直方图。这将在未来的章节中有所帮助。

介绍

在第一篇文章中,我们计算和绘制了一维直方图。它是一维的,因为我们只考虑一个特征,即像素的灰度强度值。但是在二维直方图中,您会考虑两个特征。通常在查找颜色直方图时,这两个特征是每个像素的色调和饱和度值。

有一个 python 示例(samples/python/color_histogram.py)已经用于展示如何查找颜色直方图。我们将尝试了解如何创建这样一个颜色直方图,这将有助于理解进一步的主题,如直方图反向投影。

OpenCV 中的 2D 直方图

这很简单,同样使用函数 cv2.calcHist() 进行计算。对于颜色直方图,我们需要将图像从 BGR 转换为 HSV。 (请记住,对于一维直方图,我们从 BGR 转换为灰度)。对于 2D 直方图,其参数将被修改如下:

  • channels = [0,1],因为我们需要处理 H 和 S 平面。
  • bins = [180,256],180 for H 平面,256 for S 平面。
  • range = [0,180,0,256],色调值介于 0 和 180 之间,饱和度介于 0 和 256 之间。

现在看看下面的代码:

import cv2
import numpy as np

img = cv2.imread('home.jpg')
hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
hist = cv2.calcHist([hsv], [0, 1], None, [180, 256], [0, 180, 0, 256])

Numpy 中的二维直方图

Numpy 还为此提供了一个专用函数:np.histogram2d()。 (请记住,对于一维直方图,我们使用 np.histogram())。

import cv2
import numpy as np
from matplotlib import pyplot as plt

img = cv2.imread('home.jpg')
hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
hist, xbins, ybins = np.histogram2d(h.ravel(),s.ravel(),[180,256],[[0,180],[0,256]])

第一个参数是H平面,第二个是S平面,第三个是面元的数量,第四个是他们的范围。

现在我们可以来看看如何绘制这个颜色直方图。

绘制二维直方图

方法1:使用 cv2.imshow()

我们得到的结果是一个尺寸为180x256的二维数组。所以我们可以使用cv2.imshow()函数正常显示它们。这将是一个灰度图像,它不会给出太多的颜色信息,除非你知道不同颜色的色调值。

方法2:使用 Matplotlib

我们可以使用 matplotlib.pyplot.imshow() 函数来绘制不同颜色贴图的二维直方图。它给了我们关于不同像素密度的更多信息。但是,这也不能让我们一眼看出是什么颜色,除非你知道不同颜色的色调值。不过我更喜欢这种方法。这很简单,更好。

使用此功能时,请记住,插值标志应该是 nearest,以获得更好的结果。

import cv2
import numpy as np
from matplotlib import pyplot as plt

img = cv2.imread('home.jpg')
hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
hist = cv2.calcHist( [hsv], [0, 1], None, [180, 256], [0, 180, 0, 256] )
plt.imshow(hist,interpolation = 'nearest')
plt.show()

X轴显示S值,Y轴显示色调。

在直方图中,您可以看到H = 100和S = 200附近的一些高值。它对应于天空的蓝色。同样,在H = 25和S = 100附近可以看到另一个峰值。它对应于宫殿的黄色。您可以使用任何图像编辑工具(如GIMP)来验证它。

方法3:OpenCV 示例风格

在OpenCV-Python示例(samples/python/color_histogram.py)中有一个用于颜色直方图的示例代码。

如果你运行代码,你可以看到直方图也显示相应的颜色。或者只是输出一个颜色编码的直方图。其结果是非常好的(尽管你需要添加额外的线)。

在该代码中,作者在HSV中创建了一个色彩图。然后转换成BGR。得到的直方图图像与该颜色图相乘。他还使用了一些预处理步骤去除小的孤立像素,从而得到一个很好的直方图。

我把运行代码,分析它,并自己 hack 一下的任务留给读者。

你可以在直方图中清楚地看到有什么颜色存在,蓝色在那里,黄色在那里,还有一些白色(因为图中有一个棋盘图案)在那里。