Train-YOLOv5

Automate Traning YOLOv5's Object Detection Algorthm

How to Setup

NOTE: If you don't use python as your alias for running python3.x then replace python with whatever alias you use.

Run: python setup.py

How to Run

These steps do not have to be run in order, but it will make life easier to do them in order.

1. Run python train_validate_split.py [percentage of images used for training between 0.0 - 1.0] [path to images folder] [path to labels folder]
2. Run python customData.py [file path to classes text file] [path to train validation split folder (generated by train_validate_split.py)]
3. Read yolo's tutorial.ipynb file, then run python yolo5/train.py [add all the arguments suggested by tutorial.ipynb]