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).

Classification of Microscopic Laser Engraving Surface Defect Images Based on Transfer Learning Method

A Novel CNN Ensemble Framework for Bearing Surface Defects Classification Based on Transfer Learning

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

Classification Method of Surface Defects of Aluminum Profile Based on Transfer Learning

A Deep Learning-Based Surface Defect Inspection System Using Multiscale and Channel-Compressed Features

Laser ultrasonic inspection of surface and subsurface defects in materials based on deep learning

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

Defect detection of injection molding products on small datasets using transfer learning

Classification of Surface Defects on Steel Sheet Using Convolutional Neural Networks

Transfer Learning Based Screen Defect Classification

Center-based Transfer Feature Learning With Classifier Adaptation for surface defect recognition

A Novel Automatic Classification Approach for Micro-Flaws on the Large-Aperture Optics Surface Based on Multi-Light Source Fusion and Integrated Deep Learning Architecture

TL-SDD: A Transfer Learning-Based Method for Surface Defect Detection with Few Samples

A Surface Defect Detection Based On Convolutional Neural Network

Defect classification on limited labeled samples with multiscale feature fusion and semi-supervised learning

A Light-Weight Deep-Learning Model with Multi-Scale Features for Steel Surface Defect Classification

Industrial Laser Welding Defect Detection and Image Defect Recognition Based on Deep Learning Model Developed

Automatic Detection And Classification Of Steel Surface Defect Using Deep Convolutional Neural Networks

A Weakly Supervised Surface Defect Detection Based on Convolutional Neural Network

A robust weakly supervised learning of deep Conv-Nets for surface defect inspection

Recognition Of Surface Defects On Steel Sheet Using Transfer Learning