SD-YOLO: a lightweight model for small target defect detection in complex environments of lithium batteries in new energy vehicles
摘要
Lithium batteries are pivotal in new energy vehicle (NEV) applications. This study proposes a novel defect detection model, Small-object Detection-YOLO (SD-YOLO), based on YOLOv11, to overcome the limitations of conventional methods in detecting small surface defects in lithium batteries, particularly addressing the challenge of high miss rates for low-contrast targets. The Similarity-Aware Activation Module (SimAM) is integrated into the backbone, enabling the model to prioritize small-scale defects and enhance feature extraction. The C3k2 GhostDynamicConv module is introduced to replace conventional convolution layers, enhancing feature representation while reducing computational complexity through dynamic convolutions and Ghost feature integration. To further refine model performance, the Focaler-Iou loss function is employed, replacing the original CIoU loss function, optimizing bounding box regression, and accelerating model convergence. Experimental results show that the proposed model significantly outperforms the original YOLOv11 in key performance metrics, with a 1.3% increase in precision, a 0.5% increase in recall, and a 2.2% increase in mean average precision (mAP). Furthermore, the model outperforms other state-of-the-art methods in both detection speed (FPS) and small-target detection capabilities.