Abstract:Defect detection is crucial in the quality control of industrial applications. Existing supervised methods are heavily reliant on the large amounts of labeled data. However, labeled data in some specific fields are still scarce, and it requires professionals to do expensive manual annotations. In this paper, we construct a novel self-supervised steel surface defect detection model by learning better embedding feature representation of the defect on large amounts of unlabeled data, which can achieve excellent results in downstream detection tasks. Commonly used image embeddings strategies in self-supervised contrastive learning methods destroy the spatial structures of the image and are not suitable for pre-training of object detection. To address the aforementioned issue, we preserve convolutional feature maps to mine robust data structures and local features, which can enhance the representation capability of the upstream model and make it applicable for transfer to object detection tasks. Besides, in order to eliminate the effect of random augmentations of contrastive learning, which can introduce noise on multi-target coexistence datasets, the Earth Mover's Distance (EMD) metric is employed to evaluate the contrastive matching similarity. Finally, a Self-supervised Contrastive Representation Learning framework with EMD (SCRL-EMD) is constructed through learning on large-scale unlabeled data and then transferred to Faster R-CNN and RetinaNet for detection performance validation on two public steel defect datasets. Comparative experimental results show that our method can achieve superior results than the state-of-the-art approaches. Compared to the baseline model, it achieves 4.1% and 6.8% mAP improvement on the two datasets, respectively. More importantly, a further improvement can be achieved on a smaller downstream dataset, revealing the meaningful potential of our method in exploiting more readily available unlabeled data.

Steel surface defect detection based on sparse global attention transformer

Multiscale Local and Global Feature Fusion for the Detection of Steel Surface Defects

Defect Detection in Steel Using a Hybrid Attention Network

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

A deep learning model for steel surface defect detection

AEGLR-Net: Attention Enhanced Global-Local Refined Network for Accurate Detection of Car Body Surface Defects

DAssd-Net: A Lightweight Steel Surface Defect Detection Model Based on Multi-Branch Dilated Convolution Aggregation and Multi-Domain Perception Detection Head

A steel surface defect inspection approach towards smart industrial monitoring

Steel strip surface defect detection based on multiscale feature sensing and adaptive feature fusion

Efficient Fused-Attention Model for Steel Surface Defect Detection

Multi-scale sensing and multi-dimensional feature enhancement for surface defect detection of hot-rolled steel strip

A New Steel Defect Detection Algorithm Based on Deep Learning

A steel defect detection method based on edge feature extraction via the Sobel operator

Surface defect detection and classification of steel using an efficient Swin Transformer

Steel surface defect detection based on self-supervised contrastive representation learning with matching metric

An efficient and real-time steel surface defect detection method based on single-stage detection algorithm

An End-to-End Steel Surface Defect Detection Approach via Fusing Multiple Hierarchical Features

A lightweight image-level segmentation method for steel surface defects based on cross-layer feature fusion

Steel Surface Defect Detection Using Improved Deep Learning Algorithm: ECA-SimSPPF-SIoU-Yolov5

Efficient Detection Model of Steel Strip Surface Defects Based on YOLO-V7

Dense Attention-Guided Cascaded Network for Salient Object Detection of Strip Steel Surface Defects