def image_hist2d(image_path: str): # 二维直方图 img = cv.imread(image_path, cv.IMREAD_COLOR) cv.imshow('img', img) # 图像转HSV颜色空间 hsv = cv.cvtColor(img, cv.COLOR_BGR2HSV) hist = cv.calcHist([hsv], [0, 1], None, [48, 48], [0, 180, 0, 256]) dst = cv.resize(hist, (400, 400)) # 像素归一化 cv.normalize(dst, dst, 0, 255, cv.NORM_MINMAX) # 色彩填充 dst = cv.applyColorMap(np.uint8(dst), cv.COLORMAP_JET) cv.imshow('hist', dst) plt.imshow(hist, interpolation='nearest') plt.title('2D Histogram') plt.show() cv.waitKey(0) cv.destroyAllWindows()
时间: 2024-04-19 22:23:33 浏览: 152
这段代码实现了计算并绘制图像的二维直方图。具体步骤如下:
1. 读取图像:使用`cv.imread(image_path, cv.IMREAD_COLOR)`函数读取指定路径的图像,并将其存储在变量`img`中。
2. 显示图像:使用`cv.imshow('img', img)`函数显示原始图像。
3. 转换颜色空间:使用`cv.cvtColor(img, cv.COLOR_BGR2HSV)`函数将图像从BGR颜色空间转换为HSV颜色空间,并将转换后的图像存储在变量`hsv`中。
4. 计算二维直方图:使用`cv.calcHist([hsv], [0, 1], None, [48, 48], [0, 180, 0, 256])`函数计算二维直方图。其中,`[hsv]`表示输入图像,`[0, 1]`表示通道索引,表示计算第0和第1个通道的直方图,`[48, 48]`表示直方图的大小为48x48,`[0, 180, 0, 256]`表示两个通道的范围分别是H通道(色调)的范围为0到180,S通道(饱和度)的范围为0到256。
5. 调整直方图尺寸:使用`cv.resize(hist, (400, 400))`函数将直方图调整为400x400的大小,并将调整后的直方图存储在变量`dst`中。
6. 像素归一化:使用`cv.normalize(dst, dst, 0, 255, cv.NORM_MINMAX)`函数将直方图像素值归一化到0到255的范围。
7. 色彩填充:使用`cv.applyColorMap(np.uint8(dst), cv.COLORMAP_JET)`函数将归一化后的直方图应用颜色映射,使其具有色彩填充效果,并将填充后的图像存储在变量`dst`中。
8. 显示直方图:使用`cv.imshow('hist', dst)`函数显示填充后的直方图。
9. 绘制直方图:使用`plt.imshow(hist, interpolation='nearest')`函数绘制二维直方图,`interpolation='nearest'`参数指定使用最近邻插值方法绘制图像。
10. 设置图像标题:使用`plt.title('2D Histogram')`函数设置图像的标题。
11. 显示绘制结果:使用`plt.show()`函数显示绘制的二维直方图。
12. 等待按键:使用`cv.waitKey(0)`函数等待用户按下任意按键。
13. 关闭窗口:使用`cv.destroyAllWindows()`函数关闭所有打开的窗口。
这段代码可以帮助我们分析图像在HSV颜色空间中的色调和饱和度分布情况,以便进行颜色相关的图像处理或分析。
阅读全文