基于改进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

表 5 本研究模型与主流模型检测结果对比

Tab.5 Comparison of proposed model detection results with mainstream models

模型 AP mAP@0.5 t/ms 划伤 压印 缺球 YOLOv3 0.586 0.813 0.981 0.806 34.39 YOLOv3-SPP 0.65 0.815 0.991 0.819 30.73 Faster-RCNN 0.468 0.690 0.984 0.714 124 YOLOv5s 0.676 0.846 0.986 0.836 8.28 本研究 0.717 0.876 0.991 0.861 14.96