Optics and Precision Engineering, Volume. 31, Issue 3, 404(2023)

Defect detection of cylindrical surface of metal pot combining attention mechanism

Jian QIAO1...2, Nengda CHEN1, Yanxiong WU3, Yang WU1 and Jingwei YANG1,* |Show fewer author(s)
Author Affiliations
  • 1School of Electrical and Mechanical Engineering and Automation, Foshan University, Foshan528000, China
  • 2Ji Hua Laboratory, Foshan5800, China
  • 3School of Physics and Optoelectronic Engineering, Foshan University, Foshan528000, China
  • show less

    To achieve the automatic and rapid detection and sorting of high-brightness reflection metal cylindrical pots, as well as break through the technical problems of slow speed and low efficiency of metal pot surface defect detection, a bi-directional feature pyramid network (BiFPN) was introduced in this study based on the YOLOX network. In addition, a lightweight feature fusion network model was devised on the basis of the attention mechanism, and the lightweight design of the computing model was realized. Meanwhile, the attention mechanism module was employed to learn the channel and space of feature information, effectively alleviating the semantic gap of multi-scale features and improving the detection precision of the model. Considering the unbalanced distribution of the learning weight of the network for difficult and easy classification samples, the classification loss function regarding the attenuation factor was determined. Comparisons of the feature fusion network, classification loss function, and attention mechanism module position ablation were conducted using the metal pot cylindrical surface defect dataset. The experimental results show that the fusion attention mechanism model can effectively identify six types of defects with different shapes, the average detection precision mAP0.5 of the test set realized 90.92%, and the detection frame rate was 30.84 FPS. Thus, cylindrical surface defects of metal pots can be identified and located, rapidly as well as with high precision, by using the proposed model.

    Tools

    Get Citation

    Copy Citation Text

    Jian QIAO, Nengda CHEN, Yanxiong WU, Yang WU, Jingwei YANG. Defect detection of cylindrical surface of metal pot combining attention mechanism[J]. Optics and Precision Engineering, 2023, 31(3): 404

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Information Sciences

    Received: Jun. 1, 2022

    Accepted: --

    Published Online: Mar. 7, 2023

    The Author Email: YANG Jingwei (mejwyang@fosu.edu.cn)

    DOI:10.37188/OPE.20233103.0404

    Topics