ASD-YOLO: An aircraft surface defects detection method using deformable convolution and attention mechanism

被引:13
作者
Huang, Bin [1 ]
Ding, Yan [1 ]
Liu, Guoliang [1 ]
Tian, Guohui [1 ]
Wang, Shanmei [2 ]
机构
[1] Shandong Univ, Sch Control Sci & Engn, Jinan 250061, Peoples R China
[2] Inspur Intelligence Technol Co LTD, Dept Intelligent Harbor & Nav, Jinan 250061, Peoples R China
关键词
Aircraft surface defects detection; Deep learning; YOLO; Flight safety; DAMAGE; IMPACT;
D O I
10.1016/j.measurement.2024.115300
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Aircraft surface defect (ASD) detection is crucial for ensuring flight safety. Addressing challenges such as largescale variations, irregular shapes, and sample imbalance in ASD detection, this paper proposes a ASD-YOLO network based on YOLOv5. ASD-YOLO incorporates several enhancements to improve recognition capabilities. Firstly, a new deformable convolutional feature extraction module (DCNC3) is designed to better learn defects of different shapes, which is combined with a global attention mechanism (GAM) to pay more attention to defects region information. Secondly, the feature representation of small defects is bolstered by the contextual enhancement module (CEM). Lastly, to alleviate sample imbalance problem, we introduce an exponential sliding average weight function (EMA-Slide). Experimental results on two datasets show improvements in mean Average Precision (mAP) by 5.7% and 3.4%, respectively, surpassing mainstream algorithms and offering novel approach to ASD detection.
引用
收藏
页数:14
相关论文
共 38 条
[31]  
Wang CY, 2024, Arxiv, DOI [arXiv:2402.13616, 10.48550/arXiv.2402.13616, DOI 10.48550/ARXIV.2402.13616]
[32]   YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors [J].
Wang, Chien-Yao ;
Bochkovskiy, Alexey ;
Liao, Hong-Yuan Mark .
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR, 2023, :7464-7475
[33]  
Wei Y., 2023, Mod. Aut. Technol, P106
[34]   Tiny object detection with context enhancement and feature purification [J].
Xiao, Jinsheng ;
Guo, Haowen ;
Zhou, Jian ;
Zhao, Tao ;
Yu, Qiuze ;
Chen, Yunhua ;
Wang, Zhongyuan .
EXPERT SYSTEMS WITH APPLICATIONS, 2023, 211
[35]   Aircraft visual inspection: A systematic literature review [J].
Yasuda, Yuri D. V. ;
Cappabianco, Fabio A. M. ;
Martins, Luiz Eduardo G. ;
Gripp, Jorge A. B. .
COMPUTERS IN INDUSTRY, 2022, 141
[36]  
Zhang X, 2023, Vis. Comput., P1
[37]  
Zhong R, 2023, J. Civ. Aviat. Flight Univ. China, V34
[38]   Deformable ConvNets v2: More Deformable, Better Results [J].
Zhu, Xizhou ;
Hu, Han ;
Lin, Stephen ;
Dai, Jifeng .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :9300-9308