人工智能--摄像头采集图片 识别红色、蓝色、绿色

该代码实现了一个基于颜色阈值的实时图像处理程序，能够识别并提取摄像头视频流中的红色、蓝色和绿色物体。首先，定义了红、蓝、绿三种颜色的HSV阈值范围，并通过img_process函数对图像进行预处理，提取指定颜色区域。接着，cnts_draw函数用于绘制检测到的颜色轮廓，并标记其边界框。colorfind函数通过计算直方图判断图像中的主要颜色，并输出结果。程序通过OpenCV实时捕获摄像头画面，依次处理并显示各颜色通道的结果，支持通过按下ESC键退出程序。该代码适用于颜色识别与物体检测的应用场景。

import cv2

import numpy as np

from matplotlib import pyplot as plt

# 颜色阈值定义

lower_red = np.array([0, 50, 100])

upper_red = np.array([10, 255, 255])

lower_blue = np.array([100, 50, 100])

upper_blue = np.array([124, 255, 255])

lower_green = np.array([35, 50, 100])    # 新增绿色阈值

upper_green = np.array([85, 255, 255])

# 颜色BGR值

red = (0, 0, 255)

blue = (255, 0, 0)

green = (0, 255, 0)  # 新增绿色定义

cv2.namedWindow("video", cv2.WINDOW_AUTOSIZE)

cv2.resizeWindow("video", 640, 480)

def img_process(img, lower, upper):

    """根据阈值处理图像，提取阈值内的颜色"""

    kernel = np.ones((35, 35), np.uint8)

    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

    Open = cv2.morphologyEx(hsv, cv2.MORPH_OPEN, kernel)

    mask = cv2.inRange(Open, lower, upper)

    res = cv2.bitwise_and(img, img, mask=mask)

    return res

def cnts_draw(img, res, color):

    """绘制指定颜色的轮廓"""

    canny = cv2.Canny(res, 100, 200)

    contours, _ = cv2.findContours(canny, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

    if not contours:

        cv2.imshow("video", img)

        return

   

    max_cnt = max(contours, key=cv2.contourArea)

    cv2.drawContours(img, max_cnt, -1, color, 2)

    x, y, w, h = cv2.boundingRect(max_cnt)

    cv2.rectangle(img, (x, y), (x+w, y+h), color, 3)

    cv2.imshow("video", img)

def colorfind(img):

    """识别主要颜色并打印"""

    kernel = np.ones((35, 35), np.uint8)

    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

    Open = cv2.morphologyEx(hsv, cv2.MORPH_OPEN, kernel)

   

    # 计算直方图并获取主要色相

    hist = cv2.calcHist([Open], [0], None, [180], [0, 180])

    max_hue = np.argmax(hist)  # 修正为使用argmax

   

    # 判断颜色范围

    if max_hue < 10 or max_hue > 170:  # 红色范围

        print("red")

    elif 100 <= max_hue <= 124:        # 蓝色范围

        print("blue")

    elif 35 <= max_hue <= 85:          # 新增绿色范围

        print("green")

if __name__ == "__main__":

    cap = cv2.VideoCapture(0)

    while cap.isOpened():

        flag, frame = cap.read()

        if not flag:

            print("无法读取摄像头！")

            break

           

        # 处理各颜色通道

        res_red = img_process(frame, lower_red, upper_red)

        res_blue = img_process(frame, lower_blue, upper_blue)

        res_green = img_process(frame, lower_green, upper_green)  # 新增绿色处理

       

        # 绘制轮廓

        cnts_draw(frame, res_red, red)

        cnts_draw(frame, res_blue, blue)

        cnts_draw(frame, res_green, green)  # 新增绿色轮廓

       

        # 颜色识别

        colorfind(frame)

       

        if cv2.waitKey(10) == 27:

            break

    cap.release()

    cv2.destroyAllWindows()