Abstract:Visual surface defect inspection provides an important tool for product quality assessment in a wide range of industrial applications. In recent years, diverse convolutional neural network (CNN) models have been devel-oped for the high-accuracy image classification of natural objects. However, it remains a challenging task to develop deep-learning-based approaches for surface defect inspection of industrial products. The main challenge is that surface defects of industrial products occur in low probability, therefore it is impractical to build a large image dataset containing different types of defects to train/fine-tune deep CNN models to achieve satisfactory performance when applied to industrial inspection tasks. To overcome the above-mentioned problem, we build a light-weight defect classification model based on the pre-trained SqueezeNet architecture and present three effective techniques to achieve high-accuracy defect classification using very few collected defective training samples. First, we experimentally demonstrate that it is reasonable to freeze shallower convolutional layers and only fine-tune deeper layers in the last convolutional stage when there are not enough training samples. Second, we integrate batch-size independent Group Normalization (GN) into SqueezeNet to compute stable statistics based on a limited number of defective samples for training the classification model. Third, we add an auxiliary task to perform the similarity comparison of two defective samples during the training phase, significantly increasing the size of the training data and generating more distinctive defect-specific representations. The experimental results on the NEU-CLS dataset and USB-FS dataset verify the effectiveness of the proposed method. Our method can obtain accurate defect classification results when very few defective training samples are provided, i.e., about 97.69% and 82.92% average classification accuracy on the NEU-CLS and USB-FS datasets, respectively, under the 5-shot setting (5 images per defect category).

Surface Defects Classification of Hot Rolled Strip Based on Improved Convolutional Neural Network

A ResNet50-Based Method for Classifying Surface Defects in Hot-Rolled Strip Steel

Strip Steel Surface Defects Classification Based on Generative Adversarial Network and Attention Mechanism

Classification of Surface Defects on Steel Strip Images using Convolution Neural Network and Support Vector Machine

FCCNet: Surface Defects Identification of Hot Rolled Strip Based on Lightweight Convolutional Neural Network

Surface Defects Classification of Hot Rolled Strip Based on Few-shot Learning

X-SDD: A New Benchmark for Hot Rolled Steel Strip Surface Defects Detection

Hot Rolling Finishing Mill Interstand Strip Image Classification Based on Simple Convolutional Neural Network

CNN-based hot-rolled steel strip surface defects classification: a comparative study between different pre-trained CNN models

Automatic recognition of surface defects for hot-rolled steel strip based on deep attention residual convolutional neural network

Surface Defect Identification of Strip Steel Using ViT‐RepVGG

Classification of Steel Strip Surface Defects Based on Optimized ResNet18.

A deep-learning-based approach for fast and robust steel surface defects classification

Deep ensemble transfer learning-based approach for classifying hot-rolled steel strips surface defects

Surface defect detection of hot rolled steel based on multi-scale feature fusion and attention mechanism residual block

ODNet: A High Real-Time Network Using Orthogonal Decomposition for Few-Shot Strip Steel Surface Defect Classification

Strip steel surface defect detecting method combined with a multi-layer attention mechanism network

An Effective Industrial Defect Classification Method under the Few-Shot Setting Via Two-Stream Training

Recognition Of Surface Defects On Steel Sheet Using Transfer Learning

Weld Defect Images Classification with VGG16-Based Neural Network.

Intelligent Detection and Classification of Surface Defects on Cold-Rolled Galvanized Steel Strips Using a Data-Driven Faulty Model with Attention Mechanism