首页|改进YOLOv5s的黑烟车辆检测算法研究

改进YOLOv5s的黑烟车辆检测算法研究

扫码查看
针对当前视觉黑烟车辆检测精度低、小目标难以检测的问题,提出改进YOLOv5s的黑烟车辆检测算法.首先,基于公开网络数据和真实道路拍摄图像构建黑烟车辆数据集,解决数据集受限问题.其次,改进网络模型,添加预测层,提高模型对小目标的检测性能,引入坐标注意力(Coordinate Attention,C A),增强模型的特征提取能力,进一步提高检测精度.最后,改进边界框回归损失函数为GIoU,提高边界框定位精度.实验结果表明,该改进模型能够有效地检测远距离小目标,改善漏报和虚警等问题.与原始YOLOv5s模型相比,该改进模型平均检测精度(mAP)提高3.1%,黑烟类别检测精度(AP)提高4.9%,在小目标场景中表现出较强的泛化能力.
Smoky Vehicle Detection Algorithm Based on Improved YOLOv5s Research
An improved YOLOv5s algorithm of smoky vehicle detection has been established to address the issue of low resolution and too small target to test.Firstly,the smoke vehicle dataset is constructed using public network data and real road photography to solve the problem of a limited dataset.Secondly,the introduction of Coordinate has further optimized the model.In addition,the paper introduces Coordinate and improves the regression loss function of the bounding box into GIOU to improve the location precision of the bounding box.Experiments show that the proposed model is effective in detecting small targets over a long distance,and it can solve the problem of false negative and false alarm.A comparison with the original YOLOv5s model reveals an increase of 3.1%in the average detection accuracy(mAP)and a 4.9%enhancement in the detection accuracy of the black smoke category(AP),the model exhibits strong generalization ability in small target scenarios.

Deep learningYOLOv5sSmoke detectionSmall target detectionAtmospheric environment monitoring

孟艺箐、吴晓东、李冰、管嘉俊

展开 >

东北林业大学,土木与交通学院,黑龙江 哈尔滨 150040

东北林业大学,机电工程学院,黑龙江哈尔滨 150040

深度学习 YOLOv5s 黑烟车辆检测 小目标检测 大气环境监测

黑龙江省自然科学基金基项目国家重点研发计划

E20170012017YFC0803901-2

2024

环境监控与预警
江苏省环境监测中心

环境监控与预警

CSTPCD
影响因子:0.788
ISSN:1674-6732
年,卷(期):2024.16(4)
  • 2