In order to solve the challenges of low accuracy, high model complexity, and suboptimal real-time responsiveness in small-target traffic sign detection, this research introduces an innovative lightweight detection model, CM-YOLOv5s. The model integrates the CSPCM module into the backbone and neck networks to substantially strengthen the feature extraction capability, employing an optimized C3_Faster module to reduce model complexity, adding a dedicated small-target detection head to improve the recognition effect, and innovatively introducing the NWD metric to optimize the CIoU loss function for addressing localization deviations. The experimental validation shows CM-YOLOv5s attains 84.5% detection accuracy while maintaining efficient inference, with model parameters and computational costs reduced by 20.4% and 0.6 GFLOPs respectively compared to the original YOLOv5s, exhibiting superior comprehensive performance advantages over existing methods.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

CM-YOLOv5s: A Lightweight Small Target Detection Model Based on YOLOv5s for Traffic Signs

  • Xinxin Zhou,
  • Honghao Zhao,
  • Yucai Li,
  • Qingchang Xue

摘要

In order to solve the challenges of low accuracy, high model complexity, and suboptimal real-time responsiveness in small-target traffic sign detection, this research introduces an innovative lightweight detection model, CM-YOLOv5s. The model integrates the CSPCM module into the backbone and neck networks to substantially strengthen the feature extraction capability, employing an optimized C3_Faster module to reduce model complexity, adding a dedicated small-target detection head to improve the recognition effect, and innovatively introducing the NWD metric to optimize the CIoU loss function for addressing localization deviations. The experimental validation shows CM-YOLOv5s attains 84.5% detection accuracy while maintaining efficient inference, with model parameters and computational costs reduced by 20.4% and 0.6 GFLOPs respectively compared to the original YOLOv5s, exhibiting superior comprehensive performance advantages over existing methods.