首页 | 本学科首页   官方微博 | 高级检索  
     

改进YOLOv5s的黑烟车辆检测算法研究
作者姓名:孟艺箐  吴晓东  李冰  管嘉俊
作者单位:东北林业大学,土木与交通学院
基金项目:黑龙江省自然科学基金基项目(E2017001);国家重点研发计划(2017YFC0803901-2)
摘    要:针对当前视觉黑烟车辆检测精度低、小目标难以检测的问题,提出改进YOLOv5s的黑烟车辆检测算法。首先,基于公开网络数据和真实道路拍摄图像构建黑烟车辆数据集,解决数据集受限问题。其次,改进网络模型,添加预测层,提高模型对小目标的检测性能,引入坐标注意力(Coordinate Attention,CA),增强模型的特征提取能力,进一步提高检测精度。最后,改进边界框回归损失函数为GIoU,提高边界框定位精度。实验结果表明,该改进模型能够有效地检测远距离小目标,改善漏报和虚警等问题。与原始YOLOv5s模型相比,该改进模型平均检测精度(mAP)提高3.1%,黑烟类别检测精度(AP)提高4.9%,在小目标场景中表现出较强的泛化能力。

关 键 词:深度学习  YOLOv5s  黑烟车辆检测  小目标检测  大气环境监测
收稿时间:2024-05-08
修稿时间:2024-06-14
点击此处可从《环境监控与预警》浏览原始摘要信息
点击此处可从《环境监控与预警》下载免费的PDF全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号