Is it possible to combine aug_transforms with some transforms of Albumentations for Segmentation?

[muellerzr (Zachary Mueller)]

@WaterKnight

they are very slow

Probably because they’re being done on the CPU. If you can get it working as a batch transform you can utilize the GPU instead. Also Sylvain recently ported over initial work for Albumentions (but they’re not done on the GPU)

See here:

Using albumentations with fastai v2

I added examples on how to use albumentations in the pets tutorial (still incomplete, but this in the first section that is over. (In general, the tutorial shows how to write your own transform.) Just be mindful that if you need to transform your targets as well as in your inputs (like in segmentation or object detection), you need to use a wrapper that is an ItemTransform (the tutorial explains why) and that will only work for this kind of task. It’s just 4 lines of codes to copy though, so no…