search

PeterH0323 / Smart_Construction

Base on YOLOv5 Head Person Helmet Detection on Construction Sites，基于目标检测工地安全帽和禁入危险区域识别系统，🚀😆附 YOLOv5 训练自己的数据集超详细教程🚀😆2021.3新增可视化界面❗❗
GNU General Public License v3.0
2.24k stars 469 forks source link

训练设备 #26

Closed xwq-jfl closed 3 years ago

xwq-jfl commented 3 years ago

请问训练的时候怎么指定gpu设备

PeterH0323 commented 3 years ago

通过 --device 参数进行 GPU 的选择