Module providing common Brain MRI Augmentation Methods for PyTorch.

Note

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

brainrise.transform.affine_flow(affine, shape)[source]¶

Generates a flow field given an affine matrix.

Parameters

affine : Tensor (4, 4)

an affine transform.

shape : tuple

the target output image shape.

Returns

flow : Tensor

the generated affine flow field.