Autocorrelation-Aware Aggregation Network for Salient Object Detection of Strip Steel Surface Defects
Wenqi Cui, Kechen Song, Hu Feng, Xiujian Jia, Shaoning Liu, Yunhui Yan
Abstract
In recent years, salient object detection has made great progress in natural scene images (NSIs), but salient object detection of strip steel defect images (SDIs) in industrial scenes is still an open and challenging problem. Existing detection methods are difficult to segment different types of defects with clutter and shallow contrast. Therefore, we propose a novel Autocorrelation Aware Aggregation Network (A3Net) for salient object detection of strip steel surface defects. Firstly, we use a general residual attention mechanism to enhance the encoder features and accelerate the convergence of the model. In the decoder stage, we propose a global autocorrelation module (GAM) to explore semantic information cues of high-level features to locate and guide low-level information. Then, we deploy the scale interaction module (SIM) to realize the fusion and interaction of feature information between different layers. Finally, we design a local autocorrelation module (LAM) to further refine the edge details of salient objects. We conduct detailed and rich experiments on the public strip steel surface defects dataset, which proves that our method is consistently superior to the state-of-the-art methods. In addition, we build a new challenging strip SDI dataset with multiple defect types for SOD task, which contains 4800 images with pixel-level annotations. Our dataset and code are available at: https://github.com/VDT-2048/A3Net.