基于改进YOLOv5的推力球轴承表面缺陷检测算法

袁天乐,袁巨龙,朱勇建,郑翰辰

Surface defect detection algorithm of thrust ball bearing based on improved YOLOv5

Tian-le YUAN,Ju-long YUAN,Yong-jian ZHU,Han-chen ZHENG

表 1 YOLOv5s和YOLOv5m的轴承检测结果对比

Tab.1 Comparison of YOLOv5s and YOLOv5m detection result of bearing

模型 AP p r mAP@0.5 t/ms 划伤 压印 缺球 YOLOv5s 0.567 0.789 0.988 0.855 0.757 0.782 8.28 YOLOv5m 0.550 0.807 0.991 0.811 0.783 0.783 15.64