基于yolov8，训练一个安全帽识别的目标检测模型

[@wusuopu]

很棒 很棒，我们要做一个自动收集网球的机器人，要部署一个yolo-tiny 模型，没想到还可以在windows 下训练参数。

[@devenz6994]

视频很用心 棒！

[@罐頭-v6i]

想請問小黑如果我辨識完安全帽後希望它可以輸出1分，辨識到兩葛安全帽結果輸出2分，這樣的程式可以怎麼寫?

[@Pop-dod]

怎么不更新yolo的内容了啊大佬

[@user-muddy]

你的 yml檔中沒有寫給模型label資訊的路徑感覺會出錯

[@jack-vz2dm]

会自动从官网下载

[@brave870307]

請問如何使用GPU運行

[@Lok-og1ir]

model.predict()有一個參數叫device ，輸入0/1/2代表使用gpu0/gpu1/gpu2

[@jack-vz2dm]

参数添加
device="cuda"