Opencv识别图片颜色并绘制轮廓实现代码示例_Python

一、实验原理

1、颜色空间转换：

opencv识别图片颜色的逻辑

opencv 的默认颜色空间为 bgr（蓝、绿、红），但识别特定颜色更适合在 hsv（色调、饱和度、亮度）颜色空间中进行。
hsv 优势：能够更方便地描述颜色范围，用于筛选特定颜色区域。

2、颜色范围过滤：

不同颜色的hsv值

利用 cv2.inrange 函数，通过设定的颜色范围，将指定颜色区域提取为二值掩膜。

3、图像处理：

①滤波：

通过中值滤波去除噪点，平滑图像。滤波是应用卷积来实现的，卷积的关键就是卷积核，下图为卷积核滤波原理：

②形态学变换：

通过开运算（先腐蚀后膨胀）进一步消除小噪声和孤立点。

4、轮廓提取和绘制：

利用 cv2.findcontours 提取二值图像的轮廓。

根据轮廓面积进行过滤，保留符合条件的目标区域。
使用 cv2.drawcontours 在原图上绘制轮廓。

cv2.drawcontours(image, contours, contouridx, color, thickness, linetype, hierarchy, maxlevel, offset)

二、实验代码

import cv2
import numpy as np

# 1. 输入图片并调整大小
img = cv2.imread("./color_1.png")
if img is none:
    print("无法加载图片，请检查路径！")
    exit()
img = cv2.resize(img, (0, 0), fx=0.7, fy=0.7)

# 2. 转换 hsv 颜色空间
img_hsv = cv2.cvtcolor(img, cv2.color_bgr2hsv)

# 3. 定义颜色范围并生成二值掩膜
yellow_min = np.array([26, 43, 46])  # 黄色下界
yellow_max = np.array([34, 255, 255])  # 黄色上界
img_mask = cv2.inrange(img_hsv, yellow_min, yellow_max)

# 4. 中值滤波去噪
img_blur = cv2.medianblur(img_mask, 7)

# 5. 形态学变换 - 开运算
kernel = cv2.getstructuringelement(cv2.morph_ellipse, (5, 5))
img_open = cv2.morphologyex(img_blur, cv2.morph_open, kernel)

# 6. 轮廓提取
contours, _ = cv2.findcontours(img_open, cv2.retr_external, cv2.chain_approx_simple)

# 7. 遍历轮廓并绘制
img_result = img.copy()
for contour in contours:
    area = cv2.contourarea(contour)
    if area < 200 or area > 200000:  # 根据面积过滤无效轮廓
        continue
    cv2.drawcontours(img_result, [contour], -1, (0, 0, 255), 2)  # 绘制轮廓

# 8. 显示结果
cv2.imshow("original image", img)
cv2.imshow("filtered contours", img_result)

cv2.waitkey(0)
cv2.destroyallwindows()