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

Automated defect inspection of LED chip using deep convolutional neural network

PCBNet: A Lightweight Convolutional Neural Network for Defect Inspection in Surface Mount Technology

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

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

Fusion of multi-light source illuminated images for effective defect inspection on highly reflective surfaces

A Deep Learning-Based Approach for the Automated Surface Inspection of Copper Clad Laminate Images

A Smart Surface Inspection System Using Faster R-CNN in Cloud-Edge Computing Environment

A Compact Convolutional Neural Network for Surface Defect Inspection.

Solar Cells Surface Defects Detection Based on Deep Learning

Fully convolutional networks for chip-wise defect detection employing photoluminescence images

Customized Convolutional Neural Networks Technology for Machined Product Inspection

Automated Detection of Defects with Low Semantic Information in X-ray Images Based on Deep Learning

An Efficient Network for Surface Defect Detection

Inspection and Classification of Semiconductor Wafer Surface Defects Using CNN Deep Learning Networks

Review: Application of Convolutional Neural Network in Defect Detection of 3C Products

LE–MSFE–DDNet: a defect detection network based on low-light enhancement and multi-scale feature extraction

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

Efficient Micro-LED defect detection based on microscopic vision and deep learning

TruingDet: Towards High-Quality Visual Automatic Defect Inspection for Mental Surface

An efficient lightweight convolutional neural network for industrial surface defect detection

Simultaneous Detection of Defects in Electrical Connectors Based on Improved Convolutional Neural Network