GitHub - hansalemaos/generate_training_data_ml
pip install train-generated-data-ml
$ python train_generated_data_ml.py YOURCONFIGFILE #Config file explanation here: https://github.com/hansalemaos/generate_training_data_ml
#If you don't want to use the command line, you can import the module
from train_generated_data_ml import run_yolov5_training
run_yolov5_training(YOURCONFIGFILE)