How to use Albumentations such as `ToGray` with `Ultralytics.YOLO.train`

Discussion
,
1

How can I use albumentationsx augmentations with model.train()

The documentation pages indicate that is it possible, but it does not show how/which arguments to use:

Train documentation does not mention Albumentations:

2

You just install it.

Ultralytics doesn’t let you modify the Albumentations augmentations used. It uses it with hardcoded values and transforms which are shown in the training logs after you install it.

1 Like
3

@Kallinteris-Andreas note you can view and edit the transforms directly in the code here:

So you could i.e. install in editable mode and then update these to what you see fit and train.

1 Like
4

Why are there transformations with p=0?

5

Each augmentation in Albumentations has a parameter named p that controls the probability of applying that augmentation to input data.