Skip to content

UNet Data Augmentation Tool

s184364 requested to merge Data_Augmentation_UNet into main
  • 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.

Merge request reports