自动驾驶入门（七）：基于Yolo3的红绿灯目标检测

任务介绍

除了车辆和行人检测，自动驾驶汽车在行驶过程中还必须要遵守交通法规，其中最基础的任务是红绿灯检测。

模型

红绿灯检测的模型依旧采用了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

源码链接：

https://github.com/sovit-123/Traffic-Light-Detection-Using-YOLOv3

作者：海绵宝宝