任务介绍
除了车辆和行人检测，自动驾驶汽车在行驶过程中还必须要遵守交通法规，其中最基础的任务是红绿灯检测。
模型
红绿灯检测的模型依旧采用了YOLOv3，项目代码来源于
https://github.com/sovit-123/Traffic-Light-Detection-Using-YOLOv3
项目文件结构：
其中detect.py为测试任务的入口，它会导入models.py中的Darknet模型并载入weights/下的权重文件。
测试步骤
?1. 下载预训练模型，并将模型文件best_model_12.pt放在weights/文件夹下
?链接：https://pan.baidu.com/s/1SDW7f71JmmNXRzLLMsIJVg
?提取码：88eh
?2. 下载数据集，并将数据集放在data/文件夹下并解压
链接： https://www.kaggle.com/mbornoe/lisa-traffic-light-dataset
?3. `python detect.py --source data/<name_of_your_data_file> --view-img --weights weights/best_model_12.pt --img-size 608`
运行结果展示
https://www.youtube.com/watch?v=yy3XsMFKeSg