彭晏飞, 李冬雪, 陈曦涛. Lightweight Bearing Defect Detection Method Based on Improved YOLOv5s Feature Extraction Network[J]. Mechanical Science and Technology for Aerospace Engineering, 2025, 44(10): 1785-1792. DOI: 10.13433/j.cnki.1003-8728.20230342.
改进YOLOv5s的轻量化轴承缺陷检测方法
摘要
针对现有轴承缺陷检测模型准确率低、参数量大的问题
提出一种基于YOLOv5s的轻量化轴承缺陷检测方法。首先
采用改进的EfficientViT-B0重构YOLOv5s特征提取网络
在降低模型计算量与复杂度的同时
提取更深层的语义特征信息; 其次
为解决难易样本失衡的问题
设计一种F-CIoU作为损失函数
提升检测框的定位回归精度与鲁棒性; 最后
采用基于多重注意力机制的动态头(Dynamic Head
DyHead)
强化特征语义信息
进一步优化对轴承表面损伤的分类与定位。实验结果表明
改进后的YOLOv5s的map@0.5达到了93.8%
参数量和计算量分别降低了42.3%和51.9%。该算法在保持较高精度的同时
满足工业检测部署的轻量化需求。
Abstract
Aiming at the problem of low accuracy and many parameters of the existing bearing defect detection models
a lightweight bearing defect detection method based on YOLOv5s is proposed in this paper. Firstly
the improved EfficientViT-B0 is used to reconstruct the YOLOv5s feature extraction network
which can reduce the computational complexity of the model and extract deeper semantic feature information. Secondly
in order to solve the problem of difficult and easy sample imbalance
an F-CIoU is designed as a loss function to improve the positioning regression accuracy and robustness of the detection frame. Finally
a Dynamic Head (DyHead) based on multiple attention mechanisms is used to strengthen the feature semantic information and further optimize the classification and location of bearing surface damage. The experimental results show that map@0.5 of the improved YOLOv5s is 93.8%
the parameter amount and calculation amount are reduced by 42.3 % and 51.9 % respectively. The algorithm meets the lightweight requirements of industrial inspection deployment while maintaining high accuracy.