（附代码）基于Python对交通路口的红绿灯进行颜色检测

我当刘德华

已于 2022-09-17 12:29:24 修改

阅读量3.3k

点赞数

CC 4.0 BY-SA版权

文章标签： python 人工智能深度学习

于 2022-09-17 12:28:54 首次发布

本文介绍如何使用Python进行交通路口红绿灯颜色检测。通过导入相关库，读取视频，截取ROI区域，转换到HSV颜色空间，进行二值图像颜色判定，并在图像上画出检测结果，实现红绿灯的精准识别。附带了完整的代码示例，展示了检测红灯和绿灯的实际效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、导入所需要的库

import cv2
import numpy as np

2、视频读取

首先把视频读取进来，可根据自己所使用的视频，用resize调整视频的分辨大小

cap = cv2.VideoCapture('红绿灯.mp4')
while True:
    ret,frame = cap.read()
    if ret == False:
        break
    frame=cv2.resize(frame,(1720,800))

imshow()（如下图所示）

3、截取roi区域

roiColor=frame[300:450,450:700]

截取roi的区域，也就是说，为了避免多余的干扰因素我们要把红绿灯的位置给截取出来（如下图所示

截取后的roi（如下图所示）&

200万优质内容无限畅学