Search code examples
pythonyolodata-augmentationyolov8

YOLOv8 - What is the clean and correct way to specify augmentation parameter?


as the title says, how do I set parameters for augmentation while using YOLOv8? I want to use the Python SDK and not the CLI commands.

The mantainer of the repo refer several times to https://docs.ultralytics.com/modes/predict/#arguments but I don't understand where to store these parameters? Also, I found this.

[Comment section from this site](https://i.sstatic.net/4qWX2.png)

It would be nice to seperate the settings from data.yaml, but a working solution is the priority


Solution

  • In YOLOv8, the augmentation configuration can be found at ultralytics/yolo/cfg/default.yaml. You can change the YAML file directly or set the parameter in model.train() as mentioned above.