Abstract:Deep-learning-based detection methods have been widely applied to industrial defect inspection. However, directly using vanilla detection methods fails to achieve satisfying performance due to the lack of identifiable features. In this article, a novel attention-based multiscale feature fusion module (AMFF) is proposed, aiming to enhance defect features and improve defect identification by leveraging attention mechanism in the feature fusion. AMFF includes self-enhanced attention module (SEAM) and cross-enhanced attention module (CEAM). SEAM is performed on a single feature map, which first adopts multiple dilation convolutions to enrich contextual information without compromising resolution and then utilizes the intralayer attention on the current feature map. CEAM takes both the current feature map and the adjacent feature map as input to perform cross-layer attention. The adjacent feature map is modulated with the guidance of the current feature map, which is then combined with the current feature map and the output of SEAM for final prediction. AMFF is utilized in current feature fusion networks, e.g., feature pyramid network (FPN) and path aggregation FPN (PAFPN), and is further integrated into prevalent detectors to guide them to pay more attention to defects rather than the background. Extensive experiments are conducted on two real industrial datasets released by Tianchi platform, i.e., fabric and aluminum defect datasets. For each dataset, 500 images are randomly selected for test and the rest for training. The proposed AMFF is demonstrated to significantly boost defect detection accuracy with acceptable computational cost, and the real-time performance could fully satisfy practical requirements.

Cross-Attention Regression Flow for Defect Detection

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

Knowledge Distillation Method for Surface Defect Detection.

Fully Convolutional Cross-Scale-Flows for Image-based Defect Detection

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

Multi-scale Attention and Dilation Network for Small Defect Detection

A Multi-scale Normalizing Flow Network for Surface Defects Detection

NFCF: Industrial Surface Anomaly Detection with Normalizing Flow Cross-Fitting Network

AEGLR-Net

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

Visual Defect Inspection of Metal Part Surface via Deformable Convolution and Concatenate Feature Pyramid Neural Networks

Sparse cross-transformer network for surface defect detection

Attention-Based Multiscale Feature Fusion for Efficient Surface Defect Detection

A Data-Flow Oriented Deep Ensemble Learning Method for Real-Time Surface Defect Inspection

A hierarchical attention detector for bearing surface defect detection

A Real-Time Surface Defect Detection System for Industrial Products with Long-Tailed Distribution

Steel surface defect detection based on sparse global attention transformer

A Simple Confidence-Supervised Model for High-Resolution Defect Recognition

Normalizing Flow-Based Industrial Complex Background Anomaly Detection

PyramidFlow: High-Resolution Defect Contrastive Localization using Pyramid Normalizing Flow

A Dynamic Transformer Network With Early Exit Mechanism for Fast Detection of Multiscale Surface Defects