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.

Defect Detection Method of Aluminum Profile Surface Using Deep Self-Attention Mechanism Under Hybrid Noise Conditions

Surface Defect Detection of Aluminum Profiles Based on Multiscale and Self-Attention Mechanisms

Detection of Defects on Metal Surfaces Based on PoolFormer Auto-Encoder

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

Automatic Defect Recognition Method of Aluminium Profile Surface Defects

A Lightweight Deep Learning Network with an Optimized Attention Module for Aluminum Surface Defect Detection

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

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

Research on Recognition Technology of Aluminum Profile Surface Defects Based on Deep Learning

A hierarchical attention detector for bearing surface defect detection

Surface Defect Detection for Aerospace Aluminum Profiles with Attention Mechanism and Multi-Scale Features

Defect Detection in Steel Using a Hybrid Attention Network

Particleboard Surface Defect Inspection Based on Data Augmentation and Attention Mechanisms

Approach for Automatic Defect Detection in Aluminum Casting X-Ray Images Using Deep Learning and Gain-Adaptive Multi-Scale Retinex

Self-attention guided model for defect detection of aluminium alloy casting on X-ray image

Deformable Feature Pyramid Network for Aluminum Profile Surface Defect Detection

A Feature-Oriented Reconstruction Method for Surface-Defect Detection on Aluminum Profiles

Deep Residual Multiscale Convolutional Neural Network With Attention Mechanism for Bearing Fault Diagnosis Under Strong Noise Environment

Attention-Based Multiscale Feature Fusion for Efficient Surface Defect Detection

An efficient multiscale enhancement network with attention mechanism for aluminium defect detection

Aluminum Product Surface Defect Detection Method Based on Improved CenterNet