search

Automate-Everything-Universe / vision_trainer

Notebook containing the training method
MIT License
0 stars 0 forks source link

Add image augmentation #1

Open Dragosjosan opened 6 months ago

Dragosjosan commented 6 months ago

Here is a possible solution using a .yaml file as a training parameter. https://github.com/orgs/ultralytics/discussions/1950

Dragosjosan commented 6 months ago

A newer issue regarding this topic: https://github.com/ultralytics/ultralytics/issues/7715

Yaml source which needs to be overwritten: https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/default.yaml

Dragosjosan commented 6 months ago

Code: results = model.train(data='data.yaml', cfg='custom_hyps.yaml', epochs=100)