LAACNet: Lightweight adaptive activation convolution network-based defect detection on polished metal surfaces

被引:10
作者
Lv, Zhongliang [1 ]
Lu, Zhenyu [1 ]
Xia, Kewen [1 ]
Zuo, Hailun [1 ]
Jia, Xiangyu [1 ]
Li, Honglian [2 ]
Xu, Youwei [1 ]
机构
[1] Chongqing Univ Sci & Technol, Sch Mech & Power Engn, Chongqing 400000, Peoples R China
[2] Odesa Natl Polytech Univ, UA-65044 Odesa, Ukraine
基金
中国国家自然科学基金;
关键词
Metal polished surface; Lightweight adaptive activation convolution; Deep learning; Attention mechanism; YOLO;
D O I
10.1016/j.engappai.2024.108482
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
After the metal workpiece has been polished, there may still be small defects on the surface, which will adversely affect the quality of the product and the service life and availability of the metal in severe cases. To solve this problem, the industrial sector has been seeking more lightweight and efficient solutions. Therefore, this thesis proposes a Lightweight Adaptive Activation Convolution Network (LAACNet). Firstly, this thesis proposes a lightweight convolution module that adopts features concatenation to realize intra-channel and inter-channel information transfer and fusion. Secondly, this thesis proposes an adaptive activation convolution module with the enhanced nonlinear expression of the module, which makes the deep neural network expression more powerful. This thesis proposes a spatial channel coordinate attention module to capture the long-range dependencies between image pixels better. Finally, this thesis introduces a loss function that can optimize performance in target classification and localization tasks. Experiments were conducted on the self-made datasets Metal Surface Defect-Detection (MSD-DET) and GC10-Detection (GC10-DET), achieving Mean Average Precision 50 (mAP50) of 86.3% and 66.8%, respectively. The detection performance of this model is superior to other methods. In the ablation experiment, this thesis verified the effectiveness of each module. This thesis validated the Northeastern University-Detection (NEU-DET) dataset, achieving a mAP50 of 76.0%. The results show that LAACNet exhibits excellent robustness and generalization performance in surface defect detection. In addition, the method significantly reduces the number of model parameters, providing an effective choice for lightweight and efficient solutions.
引用
收藏
页数:16
相关论文
共 49 条
  • [31] Machine Learning-Based Imaging System for Surface Defect Inspection
    Park, Je-Kang
    Kwon, Bae-Keun
    Park, Jun-Hyub
    Kang, Dong-Joong
    [J]. INTERNATIONAL JOURNAL OF PRECISION ENGINEERING AND MANUFACTURING-GREEN TECHNOLOGY, 2016, 3 (03) : 303 - 310
  • [32] Defect Detection Methods for Industrial Products Using Deep Learning Techniques: A Review
    Saberironaghi, Alireza
    Ren, Jing
    El-Gindy, Moustafa
    [J]. ALGORITHMS, 2023, 16 (02)
  • [33] Automatic Detection and Classification of Defective Areas on Metal Parts by Using Adaptive Fusion of Faster R-CNN and Shape From Shading
    Selamet, Feyza
    Cakar, Serap
    Kotan, Muhammed
    [J]. IEEE ACCESS, 2022, 10 : 126030 - 126038
  • [34] Lightweight Context-Aware Network Using Partial-Channel Transformation for Real-Time Semantic Segmentation
    Shi, Min
    Lin, Shaowen
    Yi, Qingming
    Weng, Jian
    Luo, Aiwen
    Zhou, Yicong
    [J]. IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2024, 25 (07) : 7401 - 7416
  • [35] LMFFNet: A Well-Balanced Lightweight Network for Fast and Accurate Semantic Segmentation
    Shi, Min
    Shen, Jialin
    Yi, Qingming
    Weng, Jian
    Huang, Zunkai
    Luo, Aiwen
    Zhou, Yicong
    [J]. IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (06) : 3205 - 3219
  • [36] Improved Sobel algorithm for defect detection of rail surfaces with enhanced efficiency and accuracy
    Shi Tian
    Kong Jian-yi
    Wang Xing-dong
    Liu Zhao
    Zheng Guo
    [J]. JOURNAL OF CENTRAL SOUTH UNIVERSITY, 2016, 23 (11) : 2867 - 2875
  • [37] TRC-YOLO: A real-time detection method for lightweight targets based on mobile devices
    Wang, Guanbo
    Ding, Hongwei
    Yang, Zhijun
    Li, Bo
    Wang, Yihao
    Bao, Liyong
    [J]. IET COMPUTER VISION, 2022, 16 (02) : 126 - 142
  • [38] Wang ZX, 2020, IEEE INT CONF MULTI, DOI 10.1109/icmew46912.2020.9105997
  • [39] CBAM: Convolutional Block Attention Module
    Woo, Sanghyun
    Park, Jongchan
    Lee, Joon-Young
    Kweon, In So
    [J]. COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 : 3 - 19
  • [40] Hot-Rolled Steel Strip Surface Inspection Based on Transfer Learning Model
    Wu, Hao
    Lv, Quanquan
    [J]. JOURNAL OF SENSORS, 2021, 2021