光学精密工程2024，Vol.32Issue(11)：1746-1758,13.DOI:10.37188/OPE.20243211.1746
融合多尺度特征的蜗杆表面缺陷检测
Worm surface defect detection with fusion of multi-scale features
摘要
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 re-searched.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 YO-LOv7 algorithm.Firstly,to address the differences in defect size distribution,a progressive feature pyra-mid was incorporated to reconstruct the neck network,improving the model's multi-scale feature fusion ca-pability.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 experi-ments 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.关键词
机器视觉/表面缺陷/YOLOv7/特征金字塔Key words
machine vision/surface defect detection/YOLOv7/feature pyramid分类
计算机与自动化引用本文复制引用
王蕾,郭文平,陈欣慰,夏珉..融合多尺度特征的蜗杆表面缺陷检测[J].光学精密工程,2024,32(11):1746-1758,13.基金项目
湖北省科技厅重点研发计划(重点项目)(No.2020BAA019) （重点项目）
