设计一个识别红绿灯信号灯的Python代码

以下是一个简单的识别红绿灯信号灯的 Python 代码：

import cv2

# 读取图像
img = cv2.imread('traffic_light.jpg')

# 将图像转换为 HSV 颜色空间
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

# 设定红色和绿色的阈值范围
lower_red = np.array([, 50, 50])
upper_red = np.array([10, 255, 255])
lower_green = np.array([50, 50, 50])
upper_green = np.array([70, 255, 255])

# 根据阈值范围对图像进行二值化处理
mask_red = cv2.inRange(hsv, lower_red, upper_red)
mask_green = cv2.inRange(hsv, lower_green, upper_green)

# 计算红色和绿色像素的数量
red_pixels = cv2.countNonZero(mask_red)
green_pixels = cv2.countNonZero(mask_green)

# 判断信号灯颜色
if red_pixels > green_pixels:
    print('红灯')
else:
    print('绿灯')

这个代码使用 OpenCV 库来读取图像，并将图像转换为 HSV 颜色空间。然后，根据红色和绿色的阈值范围，对图像进行二值化处理。最后，计算红色和绿色像素的数量，并根据数量判断信号灯颜色。