UNet Data Augmentation Tool

• Implemented an Augmentation class (under qim3d/utils/augmentations.py) where the user can either choose default augmentations: None, 'light', 'moderate' or 'heavy', or add a customized augmentation.
• Customization can be changed to get **kwargs as input, otherwise should be removed (user can directly put custom transformation in qim3d.utils.Dataset() instead).
• Added new version of augmentations in UNet.ipynb.

requested review from @fima

assigned to @fima and @s184364

approved this merge request

unapproved this merge request

resolved all threads

added 2 commits

• 30357d30 - Updates to Augmentation class
• 0a3a3959 - removed unnecessary packages

Compare with previous version

added 1 commit

• a9d10ab3 - removed more unneccessary packages

Compare with previous version

approved this merge request

mentioned in commit 40fc5023

merged