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 Recognition of Wheel Hub Based on Improved Faster R-CNN

Road Defect Detection from On-Board Cameras with Scarce and Cross-Domain Data

Wheel Hub Defect Detection Based on the DS-Cascade RCNN

Detection Algorithm for Bearing Roller End Surface Defects Based on Improved YOLOv5n and Image Fusion

Recognition algorithm on bearing surface defect type

Defective Surface Detection Based on Improved Faster R-CNN

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

Approaches for Accuracy Improvement of the X-ray Image Defect Detection of Automobile Casting Aluminum Parts Based on Deep Learning

Improved Faster R-CNN Based Surface Defect Detection Algorithm for Plates

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

Detection of Micro-Defects on Irregular Reflective Surfaces Based on Improved Faster R-CNN

Aeroengine Blade Surface Defect Detection System Based on Improved Faster RCNN

Research on Surface Defect Detection Technology of Flywheel Disc Semi-Finished Products Based on Deep Learning

Enhanced Faster Region Convolutional Neural Networks for Steel Surface Defect Detection

A Novel Defect Detection Method for Ferrite Shield Surface Defects by Improved Faster R-CNN

HubNet: An E2E Model for Wheel Hub Text Detection and Recognition Using Global and Local Features

Improved YOLOv8-Based Target Precision Detection Algorithm for Train Wheel Tread Defects

Integration of Deep Learning Network and Robot Arm System for Rim Defect Inspection Application

Rail Wheel Tread Defect Detection Using Improved YOLOv3

Defect classification for tire X-ray images using convolutional neural network

Design of defect detection system for process parts based on machine vision