计算机工程与应用 ›› 2024, Vol. 60 ›› Issue (19): 190-198.DOI: 10.3778/j.issn.1002-8331.2403-0292

• 图形图像处理 • 上一篇    下一篇

基于重参数化的轻量化非机动车目标检测

马超凡,李翔,王晓霞,陈晓   

  1. 陕西科技大学  电子信息与人工智能学院,西安  710021
  • 出版日期:2024-10-01 发布日期:2024-09-30

Lightweight Non-Motor Vehicle Target Detection Based on Re-Parameterization

MA Chaofan, LI Xiang, WANG Xiaoxia, CHEN Xiao   

  1. School of Electronic Information and Artificial Intelligence, Shaanxi University of Science and Technology, Xi’an 710021, China
  • Online:2024-10-01 Published:2024-09-30

摘要: 针对非机动车检测中目标数量多易被遮挡、检测模型参数量较大且检测速度慢的问题,结合YOLOv8算法和HGNetV2网络,提出一种基于重参数化的轻量化非机动车目标检测算法RCH-YOLO。使用基于HGNetV2改进的重参数化骨干网络,减少参数量和计算量,提高检测速度;在此基础上,使用Slim-neck改进颈部网络,并在检测器中使用TGConv卷积,降低网络复杂度的同时保持准确性;采用Wise-IoU损失函数计算定位损失,加速模型收敛速度,提高对重叠目标的识别准确率。实验结果显示,相较于基准模型,RCH-YOLO算法平均检测精度提高了0.3个百分点,参数量、模型大小和GFLOPs分别降低了47.9%、44.2%和54.9%,验证了其有效性。

关键词: 轻量化, 非机动车, YOLOv8, 重参数化, HGNetV2

Abstract: Aiming at the problems that the number of targets in non-motor vehicle detection is easy to be occluded, the number of detection model parameters is large and the detection speed is slow, combining YOLOv8 algorithm and HGNetV2 network, a lightweight non-motor vehicle target detection algorithm based on re-parameterization is proposed, called RCH-YOLO. Firstly, the improved re-parameterized backbone network based on HGNetV2 is used to reduce the number of parameters and calculations and improve the detection speed. Sceondly, Slim-neck is used to improve the neck network, and TGConv convolution is used in the detector to reduce the complexity of the network while maintaining accuracy. Finally, the Wise-IoU loss function is used to calculate the positioning loss, accelerate the convergence speed of the model, and improve the recognition accuracy of overlapping targets. Experimental results show that compared with the baseline model, the average detection accuracy of the proposed algorithm is improved by 0.3 percentage points, and the number of parameters, model weight and GFLOPs are reduced by 47.9%, 44.2% and 54.9% respectively, which verifies the effectiveness of the proposed algorithm.

Key words: lightweight, non-motor vehicle, YOLOv8, re-parameterization, HGNetV2