轻量化YOLOv5s-OCG的轨枕裂纹检测算法

董超群,汪战,廖平,谢帅,荣玉杰,周靖淞

Lightweight YOLOv5s-OCG rail sleeper crack detection algorithm

Chaoqun DONG,Zhan WANG,Ping LIAO,Shuai XIE,Yujie RONG,Jingsong ZHOU

表 2 YOLOv5s基础实验和改进模块的消融实验结果

Tab.2 Ablation experiment results of YOLOv5s basic experiment and improved module

编号 模型 ODConv C3_LIGHT GFPN Precision/% Recall/% mAP@0.5/% FPS/(帧·s−1) P/106 1 YOLOv5s — — — 44.4 59.7 42.9 94 7.03 2 YOLOv5s-O √ — — 45.1 61.1 44.9 91 7.03 3 YOLOv5s-C — √ — 44.7 61.1 44.8 98 5.55 4 YOLOv5s-G — — √ 45.8 61.8 46.2 92 7.35 5 YOLOv5s-OC √ √ — 45.6 61.4 45.7 96 5.56 6 YOLOv5s-OG √ — √ 45.5 61.9 46.5 92 7.35 7 YOLOv5s-CG — √ √ 45.9 61.8 46.3 95 5.62 8 YOLOv5s-OCG √ √ √ 46.2 62.1 47.1 96 5.64