基于YOLOv11n的改进铁路工人安全穿戴检测模型
武晓春,李梓宁
Improved railway worker safety wear detection model based on YOLOv11n
Xiaochun WU,Zining LI
表 1
实验环境的参数设置
Tab.1
Parameter setting for experimental environment
参数
说明
操作系统
Ubuntu 22.04
运行内存
120 GB
显卡
GeForce RTX
4090
显存
24 GB
CPU
16 vCPU Intel(R) Xeon(R) Gold
6430
深度学习框架
Pytorch 2.1.0
CUDA版本
12.1
Python版本
3.10