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

    In industrial settings, with densely arranged and distributed industrial parts, the use of horizontal box object detection often leads to issues, such as incorrect selection, missing parts, and loss of boundary direction. In this study, we propose a rotating workpiece object detection algorithm based on an enhanced version of YOLOv5s. First, a free parameter SimAM network is introduced to prioritize crucial information without increasing the number of model parameters. This enhancement enhances feature extraction in complex backgrounds and mitigates noise interference. Second, the original complete intersection over union (CIoU) regression function is replaced with the SIoU function, which incorporates an angle factor, aligning more with the rotation box detection. Substituting the activation function with Mish further enhances the model's convergence speed and accuracy. The algorithm introduces the phase-shifting coding method and an improved HardL-Tanh activation function to realize the prediction of angle and regression angle cosine values, thereby overcoming the angle multiuniformity and boundary problems associated with the five-parameter representation method and realizing the rotation frame detection of the workpiece. Experimental results demonstrate a mean accuracy precision of 97.4%, highlighting the proposed algorithm's advantages, including smaller weight files, higher average accuracy, and reduced prediction time. These qualities align with the real-time requirements of industrial applications.

    Keywords
    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