Skip to content

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