MPFANet: A Multipath Feature Aggregation Network for Steel Surface Defect Detection

Zhongyang Li,Yichun Tai,Zhenzhen Huang,Tao Peng,Zhijiang Zhang
DOI: https://doi.org/10.1088/1361-6501/ad1daf
IF: 2.398
2024-01-13
Measurement Science and Technology
Abstract:As a crucial component of product quality surveillance in the industrial manufacturing field, surface defect detection plays a pivotal role in achieving industrial automation. Although numerous defect detection methods have been widely used in steel production, complicated defect characteristics still severely affect the accuracy and efficiency of defect detection methods. To solve this problem, we propose a multi-path feature aggregation network (MPFANet) that can significantly improve steel defect detection performance. Firstly, we design an efficient multi-path aggregation feature extraction module combining gradient path with data path propagation design paradigm as the backbone feature extraction unit based on YOLOv5. Secondly, in the branch path of the backbone, deformable convolution is introduced to expand the effective receptive field and enhance the capability of extracting irregularly distributed defect features. Furthermore, we propose a chained feature enhanced pooling module that refines feature information to obtain feature maps containing more complicated defect details in the pooling layer. Experimental results on the publicly available steel surface defect datasets show that our proposed model achieves a mean average precision (mAP) of 81.9 on NEU-DET and 72.1 on GC10-DET improving by 4.6 and 3.4 over the baseline, respectively. The proposed defect detection network can achieve a balance between accuracy and efficiency, and meet the demand in practical production. The detection speed of our work reaches 41.1 FPS and 40.0 FPS on the above two datasets, which demonstrates that our method has achieved comprehensive performance in steel surface defect detection.
engineering, multidisciplinary,instruments & instrumentation
What problem does this paper attempt to address?