使用 opencv 检测具有特定颜色的圆圈

Question

我必须使用 OpenCV 和 python 检测图像中的黄色圆圈，如第一张图所示：

一旦我检测到黄色圆圈，我必须将其突出显示，如下所示：

我是 OpenCV 的新手，所以我正在寻找一些指导或帮助。感谢所有帮助

Answer 1

这是一个可能的方法：

将图像转换为 HSV
找到 upper/lower 颜色边界并创建遮罩
使用顶点数查找等高线和过滤

我们将图像转换为 HSV，然后确定下边界和上边界以使用 cv2.inRange() 创建掩码。此步骤隔离黄色对象

image = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
lower = np.array([0, 208, 94], dtype="uint8")
upper = np.array([179, 255, 232], dtype="uint8")
mask = cv2.inRange(image, lower, upper)

接下来要确定形状，我们要找到轮廓并使用顶点数进行过滤。我们使用 cv2.arcLength() 和 cv2.approxPolyDP() 来获取顶点列表和近似轮廓。我们可以检查此列表中的条目数来确定对象的形状。例如，如果轮廓有三个个顶点，则它一定是三角形。同样，如果它有四个个顶点，它一定是一个正方形。所以对于这个图像，我们可以假设它的形状是一个圆，如果它有大于一定数量的顶点。这是结果

import numpy as np
import cv2

image = cv2.imread('1.png')
original = image.copy()
image = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
lower = np.array([0, 208, 94], dtype="uint8")
upper = np.array([179, 255, 232], dtype="uint8")
mask = cv2.inRange(image, lower, upper)

# Find contours
cnts = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
# Extract contours depending on OpenCV version
cnts = cnts[0] if len(cnts) == 2 else cnts[1]

# Iterate through contours and filter by the number of vertices 
for c in cnts:
    perimeter = cv2.arcLength(c, True)
    approx = cv2.approxPolyDP(c, 0.04 * perimeter, True)
    if len(approx) > 5:
        cv2.drawContours(original, [c], -1, (36, 255, 12), -1)

cv2.imshow('mask', mask)
cv2.imshow('original', original)
cv2.imwrite('mask.png', mask)
cv2.imwrite('original.png', original)
cv2.waitKey()

使用 opencv 检测具有特定颜色的圆圈

Detect circles with specific colors using opencv

python

opencv

image

image-processing

color-detection