LIU Zilong,PANG Ming,DOU Jianming.Improved Surface Defect Detection Model for Aluminum Profile Based on CSE-YOLO[J].Software Guide,2026,25(04):98-103. DOI: 10.11907/rjdk.251095.
Improved Surface Defect Detection Model for Aluminum Profile Based on CSE-YOLO
To address the challenge of balancing model complexity, detection speed, and accuracy in aluminum profile surface defect detection, an enhanced YOLOv5-based model named CSE-YOLO was developed. First, a novel CCFBiFPN architecture was constructed by integrating CCFF with BiFPN, which strengthened multi-scale feature fusion while reducing model complexity. Second, an SDI module incorporating hybrid GSConv was implemented in the feature fusion stage to improve feature extraction capability. Finally, an ECA attention mechanism was embedded in the backbone network to enhance detection precision and overall performance. Experimental results demonstrated that compared with the baseline YOLOv5s model, the proposed approach achieved 5% and 1.3% improvements in mAP@0.5 and mAP@0.5:0.9 metrics, respectively. Concurrently, parameter count and FLOPs were reduced by 4.3% and 4.4%, with the frame rate increasing from 140 to 143 FPS. The optimized architecture effectively balanced performance and complexity, demonstrating substantial practical value for industrial quality inspection applications.
关键词
Keywords
references
TAO X , HOU W , XU D . Overview of surface defect detection methods based on deep learning [J]. IEEE/CAA Journal of Automatica Sinica , 2021 ( 47 ): 1017 - 1034 . doi: 10.16383/j.aas.c190811 http://dx.doi.org/10.16383/j.aas.c190811
DENG G W , YOU H Q , ZHU Z S . Surface defect detection of aluminum profiles based on KCC-YOLOv [J]. Laser & Optoelectronics Progress , 2024 , 61 ( 4 ): 231 - 239 . doi: 10.3788/LOP230950 http://dx.doi.org/10.3788/LOP230950
SUN L S , WEI J X , ZHU D M , et al . Surface defect detection algorithm for aluminum profiles based on AM-YOLOv3 model [J]. Laser & Optoelectronics Progress , 2021 , 58 ( 24 ): 360 - 370 . doi: 10.3788/LOP202158.2415007 http://dx.doi.org/10.3788/LOP202158.2415007
TAN M X , PANG R M , LE Q V . EfficientDet: scalable and efficient object detection [C]// 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition , 2020 : 10778 - 10787 . doi: 10.1109/cvpr42600.2020.01079 http://dx.doi.org/10.1109/cvpr42600.2020.01079
ZHAO Y , LYU W Y , XU S L , et al . DETRs beat YOLOs on real-time object detection [C]// 2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition , 2024 : 16965 - 16974 . doi: 10.1109/cvpr52733.2024.01605 http://dx.doi.org/10.1109/cvpr52733.2024.01605
PENG Y , SONKA M , CHEN D Z . U-Net v2: rethinking the skip connections of U-Net for medical image segmentation[DB/OL] . https://arxiv.org/pdf/2311.17791 https://arxiv.org/pdf/2311.17791 . doi: 10.1109/isbi60581.2025.10980742 http://dx.doi.org/10.1109/isbi60581.2025.10980742
LI H L , LI J , WEI H B , et al . Slim-neck by GSConv: a lightweight-design for real-time detector architectures [J]. Journal of Real-Time Image Processing , 2024 , 21 ( 3 ): 62 - 74 . doi: 10.1007/s11554-024-01436-6 http://dx.doi.org/10.1007/s11554-024-01436-6
WOO S , PARK J , LEE J-Y , et al . CBAM: convolutional block attention module [C]// Proceedings of the European Conference on Computer Vision , 2018 : 3 - 19 . doi: 10.1007/978-3-030-01234-2_1 http://dx.doi.org/10.1007/978-3-030-01234-2_1
WANG Q L , WU B G , ZHU P F , et al . ECA-Net: efficient channel attention for deep convolutional neural networks [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition , 2020 : 11534 - 11542 . doi: 10.1109/cvpr42600.2020.01155 http://dx.doi.org/10.1109/cvpr42600.2020.01155
Alibaba Group Holding Limite . 2018 Guangdong industrial intelligent manufacturing big data innovation competition - intelligent algorithm competition [EB/OL]. https://tianchi.aliyun.com/competition/entrance/231682/information https://tianchi.aliyun.com/competition/entrance/231682/information .
HU J , SHEN L , SUN G . Squeeze-and-excitation networks [J]. IEEE Transactions on Pattern Analysis and Machine Intelligence , 2017 , 42 : 2011 - 2023 . doi: 10.1109/cvpr.2018.00745 http://dx.doi.org/10.1109/cvpr.2018.00745
LIU Y C , SHAO Z R , HOFFMANN N . Global attention mechanism: retain information to enhance channel-spatial interactions[DB/OL] . https://arxiv.org/pdf/2112.05561 https://arxiv.org/pdf/2112.05561 .
WANG C Y , BOCHKOVSKIY A , LIAO H M . YOLOv7: trainable bag-of-freebies sets new state-of-the-art for real-time object detectors [C]// 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition , 2023 : 7464 - 7475 . doi: 10.1109/cvpr52729.2023.00721 http://dx.doi.org/10.1109/cvpr52729.2023.00721
WANG A , CHEN H , LIU L H , et al . YOLOv10: real-time end-to-end object detection[DB/OL] . https://arxiv.org/pdf/2405.14458 https://arxiv.org/pdf/2405.14458 . doi: 10.52202/079017-3429 http://dx.doi.org/10.52202/079017-3429
SUN T Q , LIU J , SONG C , et al . Small target defect detection method for aluminum profile surface based on improved YOLOv8n [J]. Modern Manufacturing Engineering , 2024 ( 12 ): 120 - 129 .
XI L F , YAERMAIMAITI Y , LIU Y J . Surface defect detection method for aluminum profiles based on improved YOLOv5 [J]. Journal of Guangxi Normal University(Natural Science Edition) , 2024 , 42 ( 1 ): 111 - 119 .