Skip to main content

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

Project description

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.


Install the dependencies and you are ready to go!

pip install augmentation


Check out the provided notebook examples!

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for augmentation, version 0.7
Filename, size File type Python version Upload date Hashes
Filename, size augmentation-0.7-py3-none-any.whl (7.6 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size augmentation-0.7.tar.gz (4.1 kB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page