ZHOU X,HAO W J,BIAN C G,et al. Detection Method for Welding Defects of YOLOv5 Steel Pipe Based on gnConv and GAM[J]. Microelectronics & Computer,2023,40(9):29-37. doi: 10.19304/J.ISSN1000-7180.2022.0778
Citation: ZHOU X,HAO W J,BIAN C G,et al. Detection Method for Welding Defects of YOLOv5 Steel Pipe Based on gnConv and GAM[J]. Microelectronics & Computer,2023,40(9):29-37. doi: 10.19304/J.ISSN1000-7180.2022.0778

Detection Method for Welding Defects of YOLOv5 Steel Pipe Based on gnConv and GAM

  • Aiming at the problems of low detection accuracy and slow detection speed caused by small target and complex background of steel pipe welding defects, an improved YOLOv5 detection algorithm is proposed. First, recursive gated convolution gnConv is used to replace the common convolution layer in the network, which enhances the interaction ability of model space, realizes efficient feature extraction, and indirectly improves the detection speed. Secondly, the use of ASPP (Atmosphere Spatial Pyramid Pooling) module not only expands the receptive field, but also improves the detection speed. Finally, GAM (Global Attention Mechanism) is added to the prediction end of the network to further enhance feature extraction and improve detection accuracy. The experimental results show that the improved al gorithm mAP achieves 92.7%, 2.1 percentage points higher than the original algorithm, and the speed is 50.8 f/s, meeting the requirements of precision and real-time of steel pipe welding defect detection.
  • loading

Catalog

    Turn off MathJax
    Article Contents

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return