Autocorrelation-Aware Aggregation Network for Salient Object Detection of Strip Steel Surface Defects

被引:23
作者
Cui, Wenqi [1 ,2 ]
Song, Kechen [1 ,2 ]
Feng, Hu [1 ,2 ]
Jia, Xiujian [1 ,2 ]
Liu, Shaoning [1 ,2 ]
Yan, Yunhui [1 ,2 ]
机构
[1] Northeastern Univ, Natl Frontiers Sci Ctr Ind Intelligence & Syst Opt, Sch Mech Engn & Automat, Minist Educ, Shenyang 110819, Liaoning, Peoples R China
[2] Northeastern Univ, Key Lab Data Analyt & Optimizat Smart Ind, Minist Educ, Shenyang 110819, Liaoning, Peoples R China
基金
中国国家自然科学基金;
关键词
Autocorrelation context-aware; salient object detection (SOD); surface defect detection;
D O I
10.1109/TIM.2023.3290965
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
In recent years, salient object detection (SOD) has made great progress in natural scene images (NSIs), but SOD 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 SOD of strip steel surface defects. First, 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 the SOD task, which contains 4800 images with pixel-level annotations. Our dataset and code are available at https://github.com/VDT-2048/A3Net.
引用
收藏
页数:12
相关论文
共 74 条
  • [1] Achanta R, 2009, PROC CVPR IEEE, P1597, DOI 10.1109/CVPRW.2009.5206596
  • [2] Salient Object Detection: A Benchmark
    Borji, Ali
    Cheng, Ming-Ming
    Jiang, Huaizu
    Li, Jia
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2015, 24 (12) : 5706 - 5722
  • [3] Chen ZY, 2020, AAAI CONF ARTIF INTE, V34, P10599
  • [4] A tutorial on the cross-entropy method
    De Boer, PT
    Kroese, DP
    Mannor, S
    Rubinstein, RY
    [J]. ANNALS OF OPERATIONS RESEARCH, 2005, 134 (01) : 19 - 67
  • [5] Cross-Scale Edge Purification Network for salient object detection of steel defect images
    Ding, Tuo
    Li, Gongyang
    Liu, Zhi
    [J]. MEASUREMENT, 2022, 199
  • [6] Dosovitskiy A, 2021, Arxiv, DOI arXiv:2010.11929
  • [7] Fan DP, 2018, Arxiv, DOI arXiv:1805.10421
  • [8] Concealed Object Detection
    Fan, Deng-Ping
    Ji, Ge-Peng
    Cheng, Ming-Ming
    Shao, Ling
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2022, 44 (10) : 6024 - 6042
  • [9] Structure-measure: A New Way to Evaluate Foreground Maps
    Fan, Deng-Ping
    Cheng, Ming-Ming
    Liu, Yun
    Li, Tao
    Borji, Ali
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 4558 - 4567
  • [10] Cross Position Aggregation Network for Few-Shot Strip Steel Surface Defect Segmentation
    Feng, Hu
    Song, Kechen
    Cui, Wenqi
    Zhang, Yiming
    Yan, Yunhui
    [J]. IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2023, 72