轻量化YOLOv5s-OCG的轨枕裂纹检测算法
董超群,汪战,廖平,谢帅,荣玉杰,周靖淞

Lightweight YOLOv5s-OCG rail sleeper crack detection algorithm
Chaoqun DONG,Zhan WANG,Ping LIAO,Shuai XIE,Yujie RONG,Jingsong ZHOU
表 1 实验硬件与软件环境
Tab.1 Experimental hardware and software environment
配置项配置规格
CPUIntel(R) Xeon(R) CPU E5-2680 v4 @2.40 GHZ
GPUNVIDIA GeForce RTX3060(12 G)
内存30 G
操作系统Ubuntu18.04
深度学习框架Pytorch1.11.0、CUDA11.3
编程语言Python