-
题名基于深度学习的工业轴承缺陷检测算法研究
被引量:1
- 1
-
-
作者
张彪
荀荣科
许家忠
-
机构
哈尔滨理工大学自动化学院
黑龙江省复杂智能系统与集成重点实验室
-
出处
《仪器仪表学报》
北大核心
2025年第4期136-149,共14页
-
文摘
针对现有轴承缺陷检测算法准确率低、存在误检以及漏检现象严重的问题,为解决这些问题,提出了一种基于YOLOv8n的轴承缺陷检测算法(LASW-YOLOv8)。该算法在YOLOv8n的基础上,引入了轻量化且高效的LiteShiftHead检测头,结合SPConv、REG和CLS模块,提升了特征提取、目标框回归和类别分类的效率与准确性。此外,算法还引入了自适应旋转卷积核模块(ARConv),增强了对多方向缺陷的检测能力;颈部网络优化模块(SAF)进一步提升了特征提取效率;同时采用Inner-WIoU损失函数,通过优化边界框定位精度并增强对小目标及复杂形状缺陷的检测能力。实验结果表明,LASW-YOLOv8算法在多个性能指标上优于其他主流算法。该算法的准确率和召回率分别提升至97.2%和96.6%,相较于YOLOv8n分别提高了3.4%和4.5%。同时,mAP0.5和mAP0.5:0.95分别达到了98.9%和73.3%,并且在运行速度上实现了83 fps。这些结果充分证明了所提改进算法的有效性,不仅能有效减少误检和漏检现象,还满足了工业检测对高准确率和实时性的要求。此外,在东北大学公共数据集(NEU-DET)的实验中,LASW-YOLOv8算法在准确率、召回率、mAP0.5和mAP0.5:0.95这4个关键指标上均表现最佳,分别为79.3%、79.9%、84.1%和49.1%,优于其他主流算法。这一表现证明了LASW-YOLOv8算法具有出色的泛化能力和鲁棒性。
-
关键词
轴承缺陷检测
LASW-YOLOv8
liteshifthead
ARConv
Inner-WIoU损失函数
-
Keywords
bearing defect detection
LASW-YOLOv8
liteshifthead
ARConv
Inner-WIoU loss function
-
分类号
TH701
[机械工程—精密仪器及机械]
TP391.4
[自动化与计算机技术—计算机应用技术]
-