Data-Driven Semantic Segmentation Method for Detecting Metal Surface Defects
Zhao Zhang, Weibo Wang, Xiaoyan Tian, Jiubin Tan
Abstract
Accurate semantic segmentation is crucial for monitoring the quality of metal surfaces in industrial production. To solve the issues of the scarce quantities and uneven distributions of metal surface defects, challenging to achieve real-time detection and hardware integration, and hard to capture boundary information, this study proposes a dual-attention multi-scale residual aggregation network, category weight calculation method, defect migration topology method, and loss calculation method for dual boundary attention. The methods solved the technical issues by aggregating the multi-scale information of the original image and exerting attention, changing the weight coefficients of categories, expanding the datasets using the topology of the defects of defective samples to a defect-free image, and paying dual attention to the boundaries of ground truth and predicted image. Compared to the fifteen mainstream methods and our previous work, this study achieved a favourable performance on five public datasets with 5.1 M parameters and real-time inference speed of 37.5 fps. Additionally, this study demonstrates commendable robustness in the presence of noise. Our code locates at https://github.com/zz-ux/Metal-surface-defect-detection.