GPU accelerated data augmentation
This code provides standard augmentation strategies on the GPU, as some of these can be intensive on the CPU. Our GPU translation is based on PyTorch. The current version supports both 2D and 3D data augmentation.
Installation
Install the dependencies and you are ready to go!
pip install augmentation
Usage
Check out the provided notebook examples!