YOLOv5实现红绿灯检测

我要开发同款
maniacs2023年09月11日
145阅读
开发技术Python
所属分类计算机视觉、深度学习、YOLOv5

作品详情

YOLOv5实现红绿灯检测
红绿灯检测数据集
基于约1W+的红绿灯(交通信号灯)检测数据集进行训练测试;
由于数据集相对较少,针对数据集做数据增强处理:翻转,旋转,裁剪,缩放,平移,拼接等;
配置好文件进行训练;
训练模型收敛后,yolov5s红绿灯检测的mAP指标大约mAP_0.5=0.93919;而,yolov5s05_416 mAP_0.5=0.77174左右;yolov5s05_320 mAP_0.5=0.71944左右;相比而言,yolov5s05比yolov5s差了很多,mAP掉了近20个点,这主要是因为减小输入input-size,不利于小目标检测。
声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论