Metadata-Version: 2.1
Name: augment-nd
Version: 0.1.3
Summary: Simple elastic augmentation for ND arrays.
Author-email: Jan Funke <funkej@hhmi.org>
License: MIT
Classifier: Programming Language :: Python :: 3
Requires-Python: >=3.7
Description-Content-Type: text/markdown
Provides-Extra: dev
License-File: LICENSE

# 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)
```
