使用OpenCV和HSV颜色空间精确检测图像中的黄色物体

本教程详细介绍了如何利用python和opencv库，通过转换到hsv颜色空间来精确检测图像中的黄色物体。与bgr颜色空间相比，hsv因其对色调、饱和度和亮度的分离，在颜色识别方面表现更优。文章将提供从图像加载、颜色空间转换、阈值分割到轮廓检测的完整步骤和代码示例，帮助读者高效实现特定颜色物体的识别。

图像中特定颜色物体检测的挑战与HSV的优势

在计算机视觉领域，识别图像中的特定颜色物体是常见的任务。然而，直接在BGR（蓝绿红）颜色空间中进行颜色检测往往面临挑战。BGR颜色模型将颜色表示为蓝、绿、红三原色的组合，这使得颜色的表示与光照强度紧密耦合。例如，一个深黄色的物体可能在BGR值上与一个浅棕色的物体相似，或者在不同光照下，同一种黄色物体在BGR通道上的值会发生显著变化，导致难以设定一个稳定且鲁棒的阈值范围。

为了克服这些限制，通常建议使用HSV（色相、饱和度、亮度）颜色空间进行颜色检测。HSV模型更符合人类对颜色的感知方式，其三个分量分别代表：

• H (Hue - 色相)：表示颜色的种类，如红色、黄色、绿色等，通常以角度表示（0-360度）。在OpenCV中，H通道被缩放到0-179。
• S (Saturation - 饱和度)：表示颜色的纯度或鲜艳程度，从0（灰色）到100%（纯色）。在OpenCV中，S通道范围是0-255。
• V (Value - 亮度)：表示颜色的明暗程度，从0（黑色）到100%（白色）。在OpenCV中，V通道范围是0-255。

在HSV空间中，色相（Hue）通道将颜色种类独立出来，使其受光照强度变化的影响远小于BGR。这意味着我们可以更容易地定义一个稳定的色相范围来识别特定颜色，例如黄色。

使用OpenCV和HSV检测黄色物体

以下是使用Python和OpenCV库，通过HSV颜色空间检测图像中黄色物体的详细步骤。

1. 导入必要的库

首先，我们需要导入cv2（OpenCV库）和numpy（用于处理数组）。

import cv2
import numpy as np

2. 加载图像

使用cv2.imread()函数加载目标图像。请确保图像文件与脚本位于同一目录下，或提供完整的路径。

# 替换为你的图像文件路径
image_path = 'screenshot.png'
image = cv2.imread(image_path)

# 检查图像是否成功加载
if image is None:
    print(f"错误：无法加载图像 '{image_path}'。请检查文件路径或文件名。")
    exit()

3. 将图像从BGR转换为HSV

OpenCV默认读取的图像是BGR格式。为了利用HSV的优势，我们需要将其转换为HSV颜色空间。

腾讯Effidit

腾讯AI Lab开发的AI写作助手，提升写作者的写作效率和创作体验

65

查看详情

hsv_image = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)

4. 定义黄色的HSV范围

这是检测特定颜色的关键一步。我们需要为黄色定义一个合适的下限和上限HSV值。在OpenCV中，H的范围是0-179（而不是0-360），S和V的范围是0-255。

对于黄色，一个常用的HSV范围是：

• Hue (H): 20-30 (对应于色轮上的黄色区域)
• Saturation (S): 100-255 (确保颜色足够鲜艳，避免检测到灰色或白色)
• Value (V): 100-255 (确保颜色具有一定的亮度，避免检测到黑色或过暗的区域)

请注意，这些值可能需要根据图像的具体情况（如光照、黄色深浅、相机白平衡）进行微调。

# 定义黄色的HSV范围
lower_yellow = np.array([20, 100, 100])
upper_yellow = np.array([30, 255, 255])

5. 创建颜色掩码

使用cv2.inRange()函数根据定义的HSV范围创建一个二值掩码（mask）。掩码中，属于黄色范围的像素点为白色（255），其余为黑色（0）。

# 根据HSV范围创建掩码
mask = cv2.inRange(hsv_image, lower_yellow, upper_yellow)

6. 应用掩码提取黄色区域

通过cv2.bitwise_and()函数将原始图像与掩码进行按位与操作，从而只保留图像中的黄色区域，背景变为黑色。这有助于直观地看到被检测到的黄色部分。

# 将掩码应用到原始图像，提取黄色物体，背景为黑色
result = cv2.bitwise_and(image, image, mask=mask)

7. 查找并绘制轮廓

为了识别独立的黄色物体，我们可以对掩码图像进行轮廓检测。轮廓是连接所有连续点（沿颜色或强度边界）的曲线，是物体形状识别的基础。

# 查找掩码中的轮廓
# cv2.RETR_EXTERNAL 只检测外层轮廓
# cv2.CHAIN_APPROX_SIMPLE 压缩水平、垂直和对角线段，只保留它们的端点
contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

# 复制原始图像用于绘制轮廓，避免修改原图
output_image = image.copy()
detected_objects_count = 0

if len(contours) > 0:
    # 遍历并绘制所有检测到的轮廓，同时可以根据面积进行筛选
    for contour in contours:
        area = cv2.contourArea(contour)
        if area > 50:  # 设定一个最小面积阈值，排除小的噪声点，可根据实际情况调整
            cv2.drawContours(output_image, [contour], -1, (0, 255, 0), 2) # 绿色轮廓，粗细为2
            detected_objects_count += 1

    if detected_objects_count > 0:
        print(f"检测到 {detected_objects_count} 个有效的黄色物体轮廓。")
    else:
        print("未检测到面积足够大的黄色物体。")
else:
    print("未检测到任何黄色物体

以上就是使用OpenCV和HSV颜色空间精确检测图像中的黄色物体的详细内容，更多请关注php中文网其它相关文章！