Yolo-MSAPF: Multiscale Alignment Fusion With Parallel Feature Filtering Model for High Accuracy Weld Defect Detection
Guan-Qiang Wang, Chizhou Zhang, Ming-Song Chen, Y.C. Lin, Xian-Hua Tan, Pei Liang, Yuxin Kang, Weidong Zeng, Wang Qiu
Abstract
This work aims to improve the low accuracy caused by interference information during real-time weld surface detection. Firstly, a weld surface dataset with 7580 pictures containing 8 types of defects was established. Secondly, an improved detection model named Yolo-MSAPF was designed based on Yolo-v5 model and verified by the self-established database. Finally, a real-time detection system was built to analyze the performance of the detection model in an industrial environment. The design principle of the Yolo-MSAPF is to eliminate interference information but enhance necessary features in each scale as much as possible by multi-scale alignment fusion (MSAF) with parallel features filtering (PFF) modules (i.e., MSAPF strategy). For the MSAF, not only the accuracy but the richness of fused features are guaranteed by aligning features at one level to fuse all other scales. After that, the fused features in each scale were individually filtered out in parallel spatial and channel in the PFF module. The results show that rate for the images with missing detection for 8 types of defects sharply drops from 21.47% to 1.68% when the MSAPF strategy is used. Moreover, it is worth noting that the <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">mAP</i> @0.5 of the Yolo-MSAPF model reaches 95.3%, which is similar to the Yolo-v7 model, while the number of parameters is reduced by 30.1%, compared with the baseline model. Additionally, the great ability to screen out unqualified weld is also verified in the industrial environment. Soon, code and dataset will be available at https://github.com/Luckycat518/Yolo-MSAPF.