...

https://github.com/UWARG/model-training

Software

Usage

Move or copy the 3 directories of the dataset so that it is in the dataset directory:

• C:\Users\WARG\Ultralytics\datasets\[test, train, val]

Make sure that any old datasets are out of this directory or have their test, train, val directories renamed (e.g. test_landing_pad, train-old, val0)! Hiding them in a directory underneath dataset is not sufficient (e.g. ...\datasets\landing_pad\test might still be erroneously used).

Activate the environment: Python Repository

Navigate into the training directory and run training:

cd training
python -m training

Training will take a few hours.

If training is interrupted, change the model load path:

• MODEL_RESUME_PATH = C:\Users\WARG\Ultralytics\runs\[latest training number]\last.pt

• data=MODEL_RESUME_PATH

Setup

Follow the instructions: Python Repository Autonomy Workflow Software

Install packages: pip install -r training/requirements.txt

...

datasets_dir: C:\Users\WARG\Ultralytics\datasets
weights_dir: C:\Users\WARG\Ultralytics\weights
runs_dir: C:\Users\WARG\Ultralytics\runs

Or more appropriate directories if desired.Use other directories if desired.

Usage

Move or copy the 3 directories of the dataset so that it is in the dataset directory:

• C:\Users\WARG\Ultralytics\datasets\[test, train, val]

Make sure that any old datasets are out of this directory or have their test, train, val directories renamed (e.g. test_landing_pad, train-old, val0)! Hiding them in a directory underneath dataset is not sufficient (e.g. ...\datasets\landing_pad\test might still be erroneously used).

Activate the environment: Autonomy Workflow Software

Navigate into the training directory and run training:

cd training
python -m training

Training will take a few hours.

If training is interrupted, change the model load path:

• MODEL_RESUME_PATH = C:\Users\WARG\Ultralytics\runs\[latest training number]\last.pt

• data=MODEL_RESUME_PATH

Hardware

Each epoch takes approximately 5 minutes to complete on an NVIDIA GeForce RTX 2060 with 6GB VRAM: https://www.techpowerup.com/gpu-specs/geforce-rtx-2060.c3310

...