红绿灯识别python代码

红绿灯识别是计算机视觉中一项重要的任务，下面是使用Python和OpenCV库编写的红绿灯识别代码示例：

import cv2

# 加载分类器
cascade_classifier = cv2.CascadeClassifier('haarcascade_traffic_light.xml')

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

# 图像灰度化
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 检测红绿灯
lights = cascade_classifier.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5)

# 标记红绿灯位置
for (x, y, w, h) in lights:
    cv2.rectangle(img, (x, y), (x+w, y+h), (0, 255, 0), 2)

# 显示结果
cv2.imshow('Traffic Light Detection', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

在代码中，使用`CascadeClassifier`加载训练好的分类器模型，然后使用`detectMultiScale`方法在图像中检测红绿灯，最后使用`rectangle`方法标记红绿灯的位置。运行代码后可以得到标记了红绿灯位置的图像。需要注意的是，此代码仅适用于检测静止的红绿灯，对于动态的红绿灯需要使用更高级的算法进行处理。