-
Notifications
You must be signed in to change notification settings - Fork 0
YoloV5 install and training
Gabriel S. Ferrante edited this page Aug 30, 2022
·
8 revisions
Repository cloned:
https://github.com/ultralytics/yolov5
Follow the installation tutorial !
Directory structure in YoloV5-Model folder:
-
BRA-Dataset
-
YoloV5-Model
-
yolov5 (detector folder)
-
BRA-Dataset.yaml
Download pretrained weights: Nano ,Small, Medium, Large, eXtra
Training command:
python3 train.py --img 416 --batch 16 --epochs 351 --data ../BRA-Dataset.yaml --weights yolov5s.pt