GitHub repository: tor4z/batchgenerators
Description: A framework for data augmentation for 2D and 3D image classification and segmentation