Any plans to add https://github.com/NVIDIA/DALI support? about albumentations HOT 6 OPEN

We have some interesting code to port into data augs libs like albumentations

https://github.com/pytorch/ignite/blob/f02a91a28153e08092dbb625472e0211f642923d/ignite/contrib/engines/dali.py#L27

and

pytorch/ignite#493 (comment)

Maybe we can try to build something like that here:

• ComposeOps
• generalized TransformPipeline

Certainly it wont support all your augs, only DALI ones, but this could help user to avoid copying common stuff.

@albu @ternaus what do you think ?

I can try to provide a draft PR if you are interested.

from albumentations.

From my perspective, most probably not. DALI has they own augmentation pipeline which is not compatible with albumentations API. Therefore support of DALI would require to introduce breaking changes in albumentations with doubtful outcome.

from albumentations.

I want to try it and figure it out if we can simplify DALI usage because it's really cool but really weird

from albumentations.

any progress @albu

from albumentations.

sorry, no progress here

from albumentations.

I have an additional example of the usage here - it really brings DALI usage inline with Albumentations/PyTorch Compose usage, so would be very cool to have integrated

Usability wise this code is a super handy way of dropping GPU acceleration as a nearly 'drop in' replacement to existing PyTorch/Albumentations workloads - we just need to find a home for it.

Take a look! Here is another Colab:

https://colab.research.google.com/drive/1XDRzDeIeteTPyzYbZp2XWjKtUWMLXqcg

from albumentations.