An improved YOLO-based method with lightweight C3 modules for object detection in resource-constrained environments
摘要
With the rapid advancement of deep learning algorithms, object detectors have achieved impressive performance in practical applications. An efficient detection framework is essential for performing detection tasks on devices with limited computational resources. However, current detection algorithms often face challenges due to their complexity, including numerous parameters and significant computational demands. To overcome these challenges, this paper introduces a streamlined and effective detection method. The integration of the FasterNet Block into the Cross-Stage Partial Network (C3) of the backbone reduces computational and storage demands. Additionally, by introducing cross-scale feature fusion in the neck network, the computational load and parameter requirements during inference are further decreased. Meanwhile, the dynamic head with multi-scale processing and Shape-IoU enhances detection accuracy and robustness, achieving a balance between lightweight design and performance. Compared to the original YOLOv5 models, the proposed lightweight method reduces the number of parameters by 29.4 to 43.0