Attention mechanism and texture contextual information for steel plate defects detection

被引:7
|
作者
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] Surface Defects Detection of Steel Plate Based on Visual Attention Mechanism
    Guo, Fang
    Zhao, Jianyu
    Jiang, Ping
    2017 CHINESE AUTOMATION CONGRESS (CAC), 2017, : 3174 - 3177
  • [2] 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
  • [3] Negative Focus Detection via Contextual Attention Mechanism
    Shen, Longxiang
    Zou, Bowei
    Hong, Yu
    Zhou, Guodong
    Zhu, Qiaoming
    Aw, Ai Ti
    2019 CONFERENCE ON EMPIRICAL METHODS IN NATURAL LANGUAGE PROCESSING AND THE 9TH INTERNATIONAL JOINT CONFERENCE ON NATURAL LANGUAGE PROCESSING (EMNLP-IJCNLP 2019): PROCEEDINGS OF THE CONFERENCE, 2019, : 2251 - 2261
  • [4] 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
  • [5] 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
  • [6] Sparse attention block: Aggregating contextual information for object detection
    Chen, Chunlin
    Yu, Jun
    Ling, Qiang
    PATTERN RECOGNITION, 2022, 124
  • [7] Research on Steel Surface Defects Detection Algorithms by YOLOv8 Based on Attention Mechanism
    Gao, ShiQuan
    Tian, Ying
    IAENG International Journal of Computer Science, 2024, 51 (09) : 1309 - 1315
  • [8] A Framework for Facial Expression Recognition Combining Contextual Information and Attention Mechanism
    Chen, Jianzeng
    Chen, Ningning
    JOURNAL OF INFORMATION PROCESSING SYSTEMS, 2024, 20 (04): : 535 - 549
  • [9] Fabric Defects Detection via Visual Attention Mechanism
    Li, Ning
    Zhao, Jianyu
    Jiang, Ping
    2017 CHINESE AUTOMATION CONGRESS (CAC), 2017, : 2956 - 2960
  • [10] Paper Defects Detection via Visual Attention Mechanism
    Jiang Ping
    Gao Tao
    2011 30TH CHINESE CONTROL CONFERENCE (CCC), 2011, : 5852 - 5856