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).