MSF-Net: Multi-Scale Feature Learning Network for Classification of Surface Defects of Multifarious Sizes

• Main Authors: Pengcheng Xu, Zhongyuan Guo, Lei Liang, Xiaohang Xu
• Format: Article
• Language: English
• Published: MDPI AG 2021-07-01
• Series: Sensors
• Subjects: surface defect classification; deep learning; convolutional neural network; multi-scale features; multi-size defects
• Online Access: https://www.mdpi.com/1424-8220/21/15/5125

In the field of surface defect detection, the scale difference of product surface defects is often huge. The existing defect detection methods based on Convolutional Neural Networks (CNNs) are more inclined to express macro and abstract features, and the ability to express local and small defects is insufficient, resulting in an imbalance of feature expression capabilities. In this paper, a Multi-Scale Feature Learning Network (MSF-Net) based on Dual Module Feature (DMF) extractor is proposed. DMF extractor is mainly composed of optimized Concatenated Rectified Linear Units (CReLUs) and optimized Inception feature extraction modules, which increases the diversity of feature receptive fields while reducing the amount of calculation; the feature maps of the middle layer with different sizes of receptive fields are merged to increase the richness of the receptive fields of the last layer of feature maps; the residual shortcut connections, batch normalization layer and average pooling layer are used to replace the fully connected layer to improve training efficiency, and make the multi-scale feature learning ability more balanced at the same time. Two representative multi-scale defect data sets are used for experiments, and the experimental results verify the advancement and effectiveness of the proposed MSF-Net in the detection of surface defects with multi-scale features.