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.

Installation

Install the dependencies and you are ready to go!

pip install augmentation

Usage

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.

Source Distribution

augmentation-0.7.tar.gz (4.1 kB view hashes)

Uploaded Source

Built Distribution

augmentation-0.7-py3-none-any.whl (7.6 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page