Computer Engineering and Applications ›› 2024, Vol. 60 ›› Issue (11): 173-181.DOI: 10.3778/j.issn.1002-8331.2308-0143

• Graphics and Image Processing • Previous Articles     Next Articles

Dim-YOLOv5n Dim Scene Object Detection Algorithm

ZHU Xiaotong, ZHANG Rongfen, LIU Yuhong, SUN Long   

  1. College of Big Data and Information Engineering, Guizhou University, Guiyang 550025, China
  • Online:2024-06-01 Published:2024-05-31

Dim-YOLOv5n昏暗场景目标检测算法

朱晓彤,张荣芬,刘宇红,孙龙   

  1. 贵州大学 大数据与信息工程学院,贵阳 550025

Abstract: Compared with normal lighting scenes, there are many interference factors in poorly illuminated scenes therefore image processing is more complex, and the existing dim object detection has shortcomings such as large number of parameters and low recognition accuracy. Aiming at the problems of false detection and missing detection in the target detection algorithm in dim scenes, Dim-YOLOv5n, a target detection algorithm in dim scenes based on YOLOv5n algorithm is proposed. ODconv-MobileNetV2 with omni-dimensional dynamic convolution (ODConv) is used to replace the backbone network to reduce the amount of computation and improve the detection accuracy. A more lightweight and efficient enhanced feature fusion network LigGFPN (lightweight generalized-FPN) based on RepGFPN (reparameterized generalized-FPN) method to improve the network’s feature extraction capability is designed, and on this basis, Ghost convolution is used to replace the traditional convolution to reduce the amount of model parameters. The experimental results show that, compared with the original algorithm, the detection accuracy P and recall R of the improved algorithm are increased by 5.3 percentage points and 5 percentage points, the average accuracy mAP0.5:0.95 and mAP0.5 are increased by 8.2 percentage points and 4.6 percentage points, respectively, and the improved algorithm effectively enhances detection accuracy while ensuring a relatively smaller model size.

Key words: dim image, YOLOv5n, omni-dimensional dynamic convolution (ODConv), MobileNetV2, reparameterized generalized-FPN (RepGFPN), Ghost convolution (GhostConv)

摘要: 相比于正常光照场景,照明不良昏暗场景干扰因素较多,图像处理较为复杂,且现有的昏暗目标检测,存在参数量大,识别准确率低等不足。针对昏暗场景下目标检测算法中存在误检与漏检等问题,提出以YOLOv5n算法为基础进行改进的昏暗场景目标检测算法Dim-YOLOv5n。利用嵌入全维动态卷积(omni-dimensional dynamic convolution,ODConv)的轻量化主干ODConv-MobileNetV2替换主干网络,在减少计算量的同时提高检测精度。基于RepGFPN(reparameterized generalized-FPN)方法设计更加轻量高效的LigGFPN(lightweight generalized-FPN)加强特征融合网络,以提高网络特征提取能力,并在此基础上,使用GhostConv(ghost convolution)替换传统卷积,以减少模型的参数量。实验结果表明,改进后算法与原算法相比,检测精度P和召回率R分别提高了5.3个百分点和5个百分点,平均精度均值mAP0.5:0.95和mAP0.5分别提升了8.2个百分点和4.6个百分点,改进的算法在保证模型较小的同时有效提高了检测准确率。

关键词: 昏暗图像, YOLOv5n, 全维动态卷积(ODConv), MobileNetV2, RepGFPN, GhostConv