Laser & Optoelectronics Progress, Volume. 61, Issue 14, 1415001(2024)

Enhanced Rotating Frame Industrial Part Detection Algorithm of YOLOv5s

Yaokun Wei1,2, Yunjiang Kang1、*, Danwei Wang1, Peng Zhao1, and Bin Xu1
Author Affiliations
  • 1Machinery Technology Development Co., Ltd., Beijing 100044, China
  • 2China Academy of Machinery Science and Technology Group, Beijing 100044, China
  • show less
    Figures & Tables(17)
    Structure of YOLOv5s network
    CSPNet modules. (a) CSP1_X component; (b) CSP2_X component
    Rotation box long side representation method
    Calculation of loss at the output of the rotating frame
    Background and target with similar texture and color
    SimAM module
    Head network structure with SimAM
    Diagram of SIoU regression
    CBM module replacement
    Schematic of angular periodic problem
    Training process of rotating box Head network
    Diagram of improved YOLOv5s network structure
    Data enhancement renderings
    Model training process
    Comparison of detection effects of different algorithms. (a) (b) (c) Base model; (d) (e) (f) proposed method
    • Table 1. Ablation experimental results

      View table

      Table 1. Ablation experimental results

      ModelSIoU/MishSimAMPSCmAP@0.95/%
      YOLOv5_R×××93.5
      A××94.6
      B××95.1
      C××95.7
      D×96.0
      E×96.4
      Proposed method96.7
    • Table 2. Performance comparison with mainstream model algorithms

      View table

      Table 2. Performance comparison with mainstream model algorithms

      MethodmAP@0.5 /%mAP@0.95 /%P /%R /%Weight size /MbitSpeed /(frame·s-1Time /ms
      Faster R_CNN94.693.291.190.989.31376.9
      YOLO_X95.394.693.593.129.52934.5
      YOLOv4_Tiny96.295.391.390.725.33033.3
      PPYOLOE_R95.495.092.891.531.22343.5
      YOLOv5s95.893.592.691.118.93231.2
      Proposed method97.496.794.694.014.63727.0
    Tools

    Get Citation

    Copy Citation Text

    Yaokun Wei, Yunjiang Kang, Danwei Wang, Peng Zhao, Bin Xu. Enhanced Rotating Frame Industrial Part Detection Algorithm of YOLOv5s[J]. Laser & Optoelectronics Progress, 2024, 61(14): 1415001

    Download Citation

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

    Category: Machine Vision

    Received: Sep. 4, 2023

    Accepted: Oct. 30, 2023

    Published Online: Jul. 4, 2024

    The Author Email: Yunjiang Kang (1228996495@qq.com)

    DOI:10.3788/LOP232025

    CSTR:32186.14.LOP232025

    Topics