Module providing common Brain MRI Augmentation Methods for PyTorch.

Note

This page is a reference documentation. It only explains the class signature, and not how to use it. Please refer to the gallery for the big picture.

class brainrise.spatial.RandomDeformation(max_displacement=4, alpha=3, order=3, seed=None)[source]¶

Apply dense random elastic deformation.

Reference: Khanal B, Ayache N, Pennec X., Simulating Longitudinal Brain MRIs with Known Volume Changes and Realistic Variations in Image Intensity, Front Neurosci, 2017.

__init__(max_displacement=4, alpha=3, order=3, seed=None)[source]¶

Init class.

Parameters

max_displacement : float, default 4

the maximum displacement in voxel along each dimension. Larger values generate more distorted images.

alpha : float, default 3

the power of the power-law momentum distribution. Larger values genrate smoother fields.

order : int, default 3

the order of the spline interpolation in the range [0, 5].

seed : int, default None

seed to control random number generator.