Extractor

Class containing all extraction tools used by Paint4Brains.

Overview:

class Paint4Brains.Extractor.Extractor

Extractor class for Paint4Brains.

This class contains the main extraction functions required.

Returns:

prob (np.array): Probability of each voxel being brain tissue or not.

load_pb()

Load the neural network with the appropriate weights into the Extractor class.

This function is run when the class is initialized

run(image)

Performs extraction on the given image

Runs the given image through the deepbrain network and returns a probability mask. :param image:

Args:

image (np.array): Original 3D brain mri volume

Returns:

prob (np.array): Probability of each voxel being brain tissue or not.