fcakyon
/

yolov5n-v7.0

Object Detection Transformers

computer-vision vision yolo yolov5

Model card Files Files and versions Community

fcakyon commited on Dec 17, 2022

Commit

b2f01dc

•

1 Parent(s): 25a9561

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -62,5 +62,5 @@ results.save(save_dir='results/')
 - Finetune the model on your custom dataset:
 ```bash
-yolov5 train --img 128 --batch 16 --weights fcakyon/yolov5n-v7.0 --epochs 1 --device cpu --freeze 10
 ```

 - Finetune the model on your custom dataset:
 ```bash
+yolov5 train --img 640 --batch 16 --weights fcakyon/yolov5n-v7.0 --epochs 10 --device cuda:0
 ```