/yolo

yoloの学習用リポジトリ

Primary LanguageJupyter Notebook

yolo

yoloの学習用リポジトリ

Yolo v5

https://github.com/ultralytics/yolov5

google colabのセッション切れ対策

90分対策

F12→Console

function ClickConnect(){
  console.log("600sごとに再接続");
  document.querySelector("colab-connect-button").click()
}
setInterval(ClickConnect,1000*600);

12時間対策

途中経過を保存できればいけそう.

ローカルで実行する場合

GTX1060のメモリサイズがgogle colabのGPUより小さいのでbatchサイズに注意
簡易的なメモリ量の見積もり方

実行コード

powershellを管理者モードで起動

cd D:\yolo
 .\.yolo\Scripts\Activate.ps1
cd yolov5
python .\train.py --batch 4 --epochs 100 --data 'D:\yolo\robomaster2.yaml' --name robomaster

CUDA

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork

参考文献

https://qiita.com/PoodleMaster/items/5f2cc3248c03b03821b8