如何在RGB颜色检测中使用“如果条件”?

我是OpenCV和python的新手。我正在尝试使用相机实时检测颜色。当检测到的颜色是红色,绿色或蓝色时,我想放置一个“如果条件”。如果检测到红色,则应打印“颜色为红色” 我要使用绿色和蓝色时也是如此。这是分别显示RGB颜色的代码。有谁可以帮我这个忙吗?在此先感谢:)

import cv2
import numpy as np
import matplotlib.pyplot as plt

cap = cv2.VideoCapture(0)

while True:
    _,frame = cap.read()
    hsv_frame = cv2.cvtColor(frame,cv2.COLOR_BGR2HSV)

    # Red color
    low_red = np.array([161,155,84])
    high_red = np.array([179,255,255])
    red_mask = cv2.inRange(hsv_frame,low_red,high_red)
    red = cv2.bitwise_and(frame,frame,mask=red_mask)


    # Blue color
    low_blue = np.array([94,80,2])
    high_blue = np.array([126,255])
    blue_mask = cv2.inRange(hsv_frame,low_blue,high_blue)
    blue = cv2.bitwise_and(frame,mask=blue_mask)

    # Green color
    low_green = np.array([25,52,72])
    high_green = np.array([102,255])
    green_mask = cv2.inRange(hsv_frame,low_green,high_green)
    green = cv2.bitwise_and(frame,mask=green_mask)

    # Every color except white
    low = np.array([0,42,0])
    high = np.array([179,255])
    mask = cv2.inRange(hsv_frame,low,high)
    result = cv2.bitwise_and(frame,mask=mask)

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

######## Chech if the shown object has red color or not ##########

    img_height,img_width,_=hsv_frame.shape
    for i in range(img_height):
        for j in range(img_width):
            if hsv_frame[i][j][1]>= low_red and hsv_frame[i][j][1]<=upper_red:
                print("Red found")

#     cv2.imshow("Red",red)
#     cv2.imshow("Blue",blue)
#     cv2.imshow("Green",green)

    if cv2.waitKey(1)==13:
        break

cap.release()
cv2.waitKey(1)
cv2.destroyAllWindows()

我得到的错误是

  

---> 40,如果hsv_frame [i] [j] [1]> = low_red和hsv_frame [i] [j] [1]

l2912503 回答:如何在RGB颜色检测中使用“如果条件”?

if hsv_frame[i][j][1]>= low_red and hsv_frame[i][j][1]<=upper_red:

比较两个数组,即像素(r,g,b)

所以您需要使用

  

any()(至少为1)

  

all()(必须全部为真)

考虑其像素比较,我建议您使用全部

用下面的代码替换它。

if (hsv_frame[i][j][1]>= low_red).all() and (hsv_frame[i][j][1]<=upper_red).all():
,

您似乎想计算图像是否为红色,其中redlow_redhigh_red边界定义。

您已经有了带有红色区域的蒙版,为什么不使用它呢?

countRed = cv2.countNonZero(red_mask)
if countRed > 0: #choose whatever condition you want
    print("Red found")
本文链接:https://www.f2er.com/3154269.html

大家都在问