应用技术与研究

基于yolov5的弱光环境航拍车辆检测

展开
  • 沈阳航空航天大学

网络出版日期: 2024-05-13

Aerial Vehicle Detection in Low Light Environment Based on Yolov5

Expand
  • Shenyang Aerospace University

Online published: 2024-05-13

摘要

针对yolov5检测无人机航拍图像时对弱光背景目标监测性能不佳这一问题,基于深度学习方法提出了一种改进 算法。首先对目标数据集visdrone2019进行数据归一化,提高后续训练效果。而后对yolov5引入添加了Mish激活函数的动态 卷积核和使用分布式偏移卷积替换 C3块的 C3_DSConv模块,并将上述两种卷积结构融合到 yolov5网络中;对网络嵌入 BiFormer注意力机制,提升对小目标检测精度。综上所述,最终得到 MODB-yolov5模型,实验结果证明该模型的 mAP和 recall 均有提高,检测阴影、黑暗环境中的车辆时精确度明显上升,且FPS较高,这保证了模型仍可用于快速检测或者实时监测。

本文引用格式

信博夫 . 基于yolov5的弱光环境航拍车辆检测[J]. 电脑与电信, 2024 , 1(1) : 78 -83 . DOI: 10.15966/j.cnki.dnydx.2024.z1.010

Abstract

An improved algorithm is proposed to address the problem that yolov5 detects UAV aerial images with poor monitoring performance for low light background targets. Firstly, data normalization is performed on the target dataset visdrone2019 to improve the detection effect. Then the dynamic convolution kernel with Mish activation function and the C3_DSConv module using distrib‐ uted offset convolution to replace the C3 block are introduced, and the above two convolution structures are fused into the yolov5 network; the BiFormer attention mechanism is embedded to improve the accuracy of small target detection. In summary, the MODByolov5 model is finally obtained, and the experimental results prove that the model's mAP and recall are both improved, and the ac‐ curacy of detecting vehicles in shadows and dark environments is significantly increased, and the FPS is high, which ensures that the model can still be used for rapid detection or real-time monitoring.
Options
文章导航

/