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.

Deep Gaussian Attention Network for Lumber Surface Defect Segmentation

An Accurate and Real-Time Surface Defects Detection Method for Sawn Lumber

Particleboard Surface Defect Inspection Based on Data Augmentation and Attention Mechanisms

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

AFFNet: An Attention-Based Feature-Fused Network for Surface Defect Segmentation

Detail-semantic guide network based on spatial attention for surface defect detection with fewer samples

A Sub-Region Unet for Weak Defects Segmentation with Global Information and Mask-Aware Loss.

Surface defect detection of sawn timbers based on efficient multilevel feature integration

An Adaptive Image Segmentation Network for Surface Defect Detection

Multi-scale Attention and Dilation Network for Small Defect Detection

A Pixel-Level Segmentation Convolutional Neural Network Based on Global and Local Feature Fusion for Surface Defect Detection

LSA-Net: Location and shape attention network for automatic surface defect segmentation

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

AnomalySeg: Deep Learning-Based Fast Anomaly Segmentation Approach for Surface Defect Detection

Surface Defect Detection and Semantic Segmentation with a Novel Lightweight Deep Neural Network

DDSNet: Deep Dual-Branch Networks for Surface Defect Segmentation

Dual attention deep learning network for automatic steel surface defect segmentation

A Multi-Source Data Fusion Network for Wood Surface Broken Defect Segmentation

Attention-Based Multiscale Feature Fusion for Efficient Surface Defect Detection

Latent Space Segmentation Model for Visual Surface Defect Inspection

Joining Spatial Deformable Convolution and a Dense Feature Pyramid for Surface Defect Detection