MRD-Net: An Effective CNN-Based Segmentation Network for Surface Defect Detection

Abstract

Surface defect detection is a crucial component in industrial production. Many algorithms based on computer vision have been successfully applied to surface defect detection, but several problems need to be solved in practical implementation. Firstly, surface defects of different products are diverse in shape, size, and location, which pose a challenge to the generalizability and accuracy of the algorithm. Secondly, the algorithm is required to be real-time in practical detection. To address these problems, we propose a multi-scale feature enhancement fusion and reverse attention network for surface defect detection, named MRD-Net, to achieve real-time and end-to-end defect segmentation. The framework first extracts multi-scale feature maps from the pre-trained MobileNetV2. Then the multi-scale feature enhancement fusion module is proposed to enhance and fuse the feature maps of the deeper layers in the backbone to improve detection performance. Through several skip connections, the adjacent branches are connected top-down. The reverse attention module is applied in shallow branches to utilize deeper prediction information. Finally, the boundary refinement module is added to refine the object boundary and improve prediction accuracy. Our proposed method requires only a small number of defective samples and achieves a high detection accuracy. The experimental results on three datasets (14 industrial products) show that the proposed method outperforms the three state-of-the-art segmentation methods in terms of generalizability and accuracy, it also achieves the requirement of real-time detection with a speed of 52 FPS for a 352*352 image.