Surface Defect Detection Based on Adaptive Multi-Scale Feature Fusion
Guochen Wen, Li Cheng, Haiwen Yuan, Xuan Li
Abstract
Surface defect detection plays a quality assurance role in industrial manufacturing processes. However, the diversity of defects and the presence of complex backgrounds bring significant challenges to salient object detection. To this end, this study proposes a new adaptive multi-scale feature fusion network (AMSFF-Net) to solve the SOD problem of object surface defects. The upsampling fusion module used adaptive weight fusion, global feature adaptive fusion, and differential feature adaptive fusion to fuse information of different scales and levels. In addition, the spatial attention (SA) mechanism was introduced to enhance the effective fusion of multi-feature maps. Preprocessing techniques such as aspect ratio adjustment and random rotation were used. Aspect ratio adjustment helps to identify and locate defects of different shapes and sizes, and random rotation enhances the ability of the model to detect defects at different angles. The negative samples and non-uniform-distribution samples in the magnetic tile defect dataset were further removed to ensure data quality. This study conducted comprehensive experiments, demonstrating that AMSFF-Net outperforms existing state-of-the-art technologies. The proposed method achieved an S-measure of 0.9038 and an Fβmax of 0.8782, which represents a 1% improvement in Fβmax compared to the best existing methods.