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.intensity.KDENormalize(mask=None, modality='T1w', norm_value=1)[source]¶

Use kernel density estimation to find the peak of the white matter in the histogram of a skull-stripped image. Then normalize intensitites to a normalization value.

__init__(mask=None, modality='T1w', norm_value=1)[source]¶

Init class.

Parameters

mask : array, default None

the brain mask.

modality str, default ‘T1w’

the modality (T1w, T2w, FLAIR, MD, last, largest, first).

norm_value : float, default 1

the new intensity value for the detected WM peak.