search

mb010 / AstroAugmentations

A package with various custom augmentations implemented which are specifically designed around astronomical data.
MIT License
7 stars 4 forks source link

Transformation Application #1

Closed mb010 closed 2 years ago

mb010 commented 2 years ago

Potentially update transformation handling to be in line with pytorch transforms. The suggestion being that A.Compose([]) objects are called using a lambda function before being passed into the data loader. Something like: transform = lambda x : A.Compose([...])[image=x]['image'].

mb010 commented 2 years ago

Encouraging users to use only Albumentations instead of hybrid solutions.