Как использовать условие «if» для определения цвета RGB?

Question

Я новичок в 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, 255])
    blue_mask = cv2.inRange(hsv_frame, low_blue, high_blue)
    blue = cv2.bitwise_and(frame, frame, mask=blue_mask)

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

    # Every color except white
    low = np.array([0, 42, 0])
    high = np.array([179, 255, 255])
    mask = cv2.inRange(hsv_frame, low, high)
    result = cv2.bitwise_and(frame, 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] <= upper_red: ValueError: Значение истинности массива с более чем одним элементом является неоднозначным. Используйте a.any () или a.all () </p>

Answer 1

Похоже, вы хотите сосчитать, если изображение имеет красный цвет или нет, где red определяется вашими low_red и high_red границами.

У вас уже есть маска с красной областью, почему бы не использовать ее?

countRed = cv2.countNonZero(red_mask)
if countRed > 0: #choose whatever condition you want
    print("Red found")

Answer 2

Строка

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

сравнивает два массива, т.е. пиксели (r, g, b) <= (r, g, b), которые будут возвращать 3 значения. т.е. (Истина / Ложь, Истина / Ложь, Истина / Ложь). </p>

Так что вам нужно использовать либо

any () (по крайней мере 1 верно)

или

all () (должно быть все true)

, учитывая сравнение пикселей, я бы порекомендовал вам использовать all,

замените его приведенным ниже кодом.

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