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 条
  • [1] Aghdam S. R., 2012, 2012 7th IEEE Conference on Industrial Electronics and Applications (ICIEA 2012). Proceedings, P1447, DOI 10.1109/ICIEA.2012.6360951
  • [2] Chandel Ruchika., 2013, International Journal of Advanced Research in Computer Science and Software Engineering, V3
  • [3] Online Detection of Surface Defects Based on Improved YOLOV3
    Chen, Xuechun
    Lv, Jun
    Fang, Yulun
    Du, Shichang
    [J]. SENSORS, 2022, 22 (03)
  • [4] Local keypoint-based Faster R-CNN
    Ding, Xintao
    Li, Qingde
    Cheng, Yongqiang
    Wang, Jinbao
    Bian, Weixin
    Jie, Biao
    [J]. APPLIED INTELLIGENCE, 2020, 50 (10) : 3007 - 3022
  • [5] Improvement of Lightweight Convolutional Neural Network Model Based on YOLO Algorithm and Its Research in Pavement Defect Detection
    Du, Fu-Jun
    Jiao, Shuang-Jian
    [J]. SENSORS, 2022, 22 (09)
  • [6] Elbehiery H, 2005, PROC WRLD ACAD SCI E, V5, P158
  • [7] Automatic detection and classification of manufacturing defects in metal boxes using deep neural networks
    Essid, Oumayma
    Laga, Hamid
    Samir, Chafik
    [J]. PLOS ONE, 2018, 13 (11):
  • [8] Automatic Rail Surface Defects Inspection Based on Mask R-CNN
    Guo, Feng
    Qian, Yu
    Rizos, Dimitris
    Suo, Zhi
    Chen, Xiaobin
    [J]. TRANSPORTATION RESEARCH RECORD, 2021, 2675 (11) : 655 - 668
  • [9] Han X., 2022, P ANN C CHIN EL SOC, P217
  • [10] An adaptive level-selecting wavelet transform for texture defect detection
    Han, Yanfang
    Shi, Pengfei
    [J]. IMAGE AND VISION COMPUTING, 2007, 25 (08) : 1239 - 1248