Metadata-Version: 2.1
Name: augment-nd
Version: 0.1.1
Summary: Simple elastic augmentation for ND arrays.
Home-page: https://github.com/funkey/augment
Author: Jan Funke
Author-email: jfunke@iri.upc.edu
License: MIT
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 2
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Description-Content-Type: text/markdown
Requires-Dist: h5py
Requires-Dist: numpy
Requires-Dist: scipy

# augment

A simple elastic augmentation for ND arrays.

# Installation

`pip install augment-nd`

# Usage

```
import augment
import numpy as np
import math

# create some example data
image = np.zeros((100,500,500), dtype=np.float)
image[:] = 0.5
image[:,:10,:] = 0.75
image[:10,:10,:10] = 1
image[:,::10,:] = 1
image[:,:,::10] = 1

transformation = augment.create_identity_transformation(image.shape)

# jitter in 3D
transformation += augment.create_elastic_transformation(
        image.shape,
        num_control_points = [3,10,10],
        jitter_sigma = [0.3, 10, 10])

# rotate around z axis
transformation += augment.create_rotation_transformation(
        image.shape,
        math.pi/4)

# apply transformation
image = augment.apply_transformation(image, transformation)
```


