augmentation

A library, based on PyTorch, that performs data augmentation on the GPU


License
MIT
Install
pip install augmentation==0.7

Documentation

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!