如何使用Python-OpenCV中的`cv2.inRange`检测两种不同的颜色?

24

如何定义两种不同颜色(例如红色和蓝色,因为它们在HSV颜色中不相邻)的“较低”和“较高”范围?

这个属于红色:

lower_red = np.array([160,20,70])
upper_red = np.array([190,255,255])

这一个属于蓝色:

lower_blue = np.array([101,50,38])
upper_blue = np.array([110,255,255])

我尝试使用if条件将它们组合起来,或者创建它们自己的函数,但都没有成功,你们能给我提供解决方案吗?

P / s:Python中的OpenCV


我不懂Python,但通常你需要为每种颜色创建掩码并将它们 OR 起来。 - Mark Setchell
可能会有所帮助:https://dev59.com/HFwY5IYBdhLWcg3wh4Iq#32523532 - Mark Setchell
4个回答

58

您可以使用cv2.bitwise_or来获取两个color掩模的最终掩模。

import cv2

## Read
img = cv2.imread("sunflower.jpg")

## convert to hsv
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

## mask of green (36,0,0) ~ (70, 255,255)
mask1 = cv2.inRange(hsv, (36, 0, 0), (70, 255,255))

## mask o yellow (15,0,0) ~ (36, 255, 255)
mask2 = cv2.inRange(hsv, (15,0,0), (36, 255, 255))

## final mask and masked
mask = cv2.bitwise_or(mask1, mask2)
target = cv2.bitwise_and(img,img, mask=mask)

cv2.imwrite("target.png", target)

来源:

在这里输入图片描述

寻找绿色和黄色(范围不是非常准确):

在这里输入图片描述


顺便说一句,为了获得更精确的范围,这是我相关答案中的参考地图:

如何定义阈值以仅检测图像中的绿色物体:Opencv

在这里输入图片描述


如何在RGB输入中动态输入上下范围?例如,我有一个应用程序,用户提交一张图像并选择两个RGB值,即上限和下限,以便从背景中去除。我该如何将这两个RGB值转换为适用于cv2.inrange函数的cv2友好的HSV值? - Logan Price

4
下面的图片展示了HSV颜色空间,它使用色相、饱和度和明度(或亮度)来工作。

HSV Colour Space

在使用HSV颜色空间时,重要的是要记住这一点,即红色和绿色等概念是一种回到不同数据类型的转换。
因此,您的上限和下限只能是此空间中的一个点,但可以包括红色和蓝色光谱的部分,即紫色。您需要选择符合所需处理输出标准的阈值。
或者运行两个单独的循环,第一个用于阈值化红色,第二个用于阈值化蓝色,然后使用OpenCV混合函数将两个图像混合在一起。有关混合两个颜色空间,请参见此处

我正在处理交通标志,其中一个是红色的,另一个是蓝色的,我无法将它们混合在一起。有没有办法设置两个不同的范围或者只是运行两个单独的循环? - Nhiên Ngô Đình

2
# Make a copy of the image

image_copy = np.copy(image)
## TODO: Define the color selection boundaries in RGB values
# play around with these values until you isolate the blue background

lower_blue = np.array([200,0,0]) 
upper_blue = np.array([250,250,255])

# Define the masked area

mask = cv2.inRange(image_copy, lower_blue, upper_blue)
# Vizualize the mask

plt.imshow(mask,cmap='gray')

0

或者,您可以创建两个掩码并将它们合并成一个掩码,如下所示:

finalmask = mask_red | mask_blue

这与使用 bitwise_or 相同。您可以像这样连接多个掩码;

maskf = maskA | maskB | maskC

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接