• Optics and Precision Engineering
  • Vol. 32, Issue 11, 1746 (2024)
Lei WANG, Wenping GUO*, Xinwei CHEN, and Min XIA
Author Affiliations
  • School of Optical and Electronic Information, Huazhong University of Science and Technology, Hubei430070, China
  • show less
    DOI: 10.37188/OPE.20243211.1746 Cite this Article
    Lei WANG, Wenping GUO, Xinwei CHEN, Min XIA. Worm surface defect detection with fusion of multi-scale features[J]. Optics and Precision Engineering, 2024, 32(11): 1746 Copy Citation Text show less

    Abstract

    To tackle the challenges of reliance on manual inspection, low detection efficiency, and high costs in detecting surface defects on worm gear teeth, automated methods utilizing machine vision were researched. A defect collection system was designed to capture worm gear defects, and data augmentation strategies were introduced to handle varying defect occurrence rates. Enhancements were made to the YOLOv7 algorithm. Firstly, to address the differences in defect size distribution, a progressive feature pyramid was incorporated to reconstruct the neck network, improving the model's multi-scale feature fusion capability. Secondly, an attention mechanism was added to minimize interference from non-defective areas and bolster the model's focus on defects. Lastly, the regression loss function was modified to SIOU, and orientation consideration was included during network training to boost detection accuracy. Ablation experiments demonstrated the effectiveness of these improvements. With a 20.7% reduction in parameter count, experimental results show that the proposed algorithm achieves a 3.3 percentage point increase in accuracy compared to the YOLOv7 algorithm. Additionally, when compared to other algorithms like YOLOR and YOLOv5m, this algorithm provides optimal detection performance, effectively meeting the requirement for automated detection of surface defects in worm gears.
    Lei WANG, Wenping GUO, Xinwei CHEN, Min XIA. Worm surface defect detection with fusion of multi-scale features[J]. Optics and Precision Engineering, 2024, 32(11): 1746
    Download Citation