Attention mechanism and texture contextual information for steel plate defects detection

被引:15
|
作者
Zhang, Chi [1 ]
Cui, Jian [1 ]
Wu, Jianguo [1 ]
Zhang, Xi [1 ]
机构
[1] Peking Univ, Dept Ind Engn & Management, Beijing, Peoples R China
关键词
Statistical texture feature fusion; Contextual information mining; Attention mechanism; Steel surface defect detection; CLASSIFICATION;
D O I
10.1007/s10845-023-02149-6
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In order to achieve rapid inference and generalization results, the majority of Convolutional Neural Network (CNN) based semantic segmentation models strive to mine high-level features that contain rich contextual semantic information. However, at steel plate defects detection scenario, some background textures' noises are similar to the foreground leading to hard distinguishment, which will significantly interfere with feature extraction. Texture features themselves often hold the most plentiful contextual information. Despite this, semantic segmentation tasks rarely take texture features into account when identifying surface defects on steel plates. In that case, the essential details, such as the edge texture and other intuitive low-level features, will generally cannot be included into the final feature map. To address the problems of inefficient accuracy and slow speed of existing detection, this study proposed a steel plate surface defect detection method using contextual information and attention mechanism, and utilizes a multi-layer feature extraction method and fusion framework based on low-level statistical textures. Through the identification of pixel-level spatial and correlation relationships, characteristics of low-level defects are extracted. Furthermore, to effectively incorporate statistical texture in CNN, a novel quantization technique has been developed. This quantization method allows for the conversion of continuous texture into various levels of intensity. The network parameters were iterated in a gradient direction, facilitating the defects division. Empirical results have demonstrated the feasibility of applying the proposed approach to practical steel plate testing. Additionally, ablation experiments have demonstrated that the method is capable of effectively enhancing surface defect detection for steel plates, resulting in industry-leading performance.
引用
收藏
页码:2193 / 2214
页数:22
相关论文
共 50 条
  • [1] CAIINET: Neural network based on contextual attention and information interaction mechanism for depression detection
    Zhou, Li
    Liu, Zhenyu
    Yuan, Xiaoyan
    Shangguan, Zixuan
    Li, Yutong
    Hu, Bin
    DIGITAL SIGNAL PROCESSING, 2023, 137
  • [2] Real-Time Lane Detection by Using Biologically Inspired Attention Mechanism to Learn Contextual Information
    Zhang, Lu
    Jiang, Fengling
    Kong, Bin
    Yang, Jing
    Wang, Can
    COGNITIVE COMPUTATION, 2021, 13 (05) : 1333 - 1344
  • [3] Real-Time Lane Detection by Using Biologically Inspired Attention Mechanism to Learn Contextual Information
    Lu Zhang
    Fengling Jiang
    Bin Kong
    Jing Yang
    Can Wang
    Cognitive Computation, 2021, 13 : 1333 - 1344
  • [4] Pest Region Detection in Complex Backgrounds via Contextual Information and Multi-Scale Mixed Attention Mechanism
    Zhang, Wei
    Sun, Youqiang
    Huang, He
    Pei, Haotian
    Sheng, Jiajia
    Yang, Po
    AGRICULTURE-BASEL, 2022, 12 (08):
  • [5] Geospatial Contextual Attention Mechanism for Automatic and Fast Airport Detection in SAR Imagery
    Tan, Siyu
    Chen, Lifu
    Pan, Zhouhao
    Xing, Jin
    Li, Zhenhong
    Yuan, Zhihui
    IEEE ACCESS, 2020, 8 (08): : 173627 - 173640
  • [6] Intelligent Detection of Steel Defects Based on Improved Split Attention Networks
    Hao, Zhiqiang
    Wang, Zhigang
    Bai, Dongxu
    Tao, Bo
    Tong, Xiliang
    Chen, Baojia
    FRONTIERS IN BIOENGINEERING AND BIOTECHNOLOGY, 2022, 9
  • [7] Graph attention mechanism with global contextual information for multi-label image recognition
    Ban, Xiaoxiao
    Li, Peihua
    Wang, Qilong
    Zhou, Shoujun
    Guo, Shijie
    Wang, Yuanquan
    JOURNAL OF ELECTRONIC IMAGING, 2021, 30 (06)
  • [8] DCAM-Net: A Rapid Detection Network for Strip Steel Surface Defects Based on Deformable Convolution and Attention Mechanism
    Chen, Haixin
    Du, Yongzhao
    Fu, Yuqing
    Zhu, Jianqing
    Zeng, Huanqiang
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2023, 72
  • [9] A Pedestrian Detection Network Based on an Attention Mechanism and Pose Information
    Jiang, Zhaoyin
    Huang, Shucheng
    Li, Mingxing
    APPLIED SCIENCES-BASEL, 2024, 14 (18):
  • [10] Combining Contextual Information by Self-attention Mechanism in Convolutional Neural Networks for Text Classification
    Wu, Xin
    Cai, Yi
    Li, Qing
    Xu, Jingyun
    Leung, Ho-fung
    WEB INFORMATION SYSTEMS ENGINEERING, WISE 2018, PT I, 2018, 11233 : 453 - 467