FastSurferCNN.data_loader.dataset

class FastSurferCNN.data_loader.dataset.MultiScaleDataset(dataset_path, cfg, gn_noise=False, transforms=None)[source]

Class for loading aseg file with augmentations (transforms).

Methods

get_subject_names()

Get the subject name.

unify_imgs(img, label, weight)

Pad img, label and weight.

get_subject_names()[source]

Get the subject name.

Returns:
list

List of subject names.

unify_imgs(img, label, weight)[source]

Pad img, label and weight.

Parameters:
imgnpt.NDArray

Image to unify.

labelnpt.NDArray

Labels of the image.

weightnpt.NDArray

Weights of the image.

Returns:
np.ndarray

Img.

np.ndarray

Label.

np.ndarray

Weight.

class FastSurferCNN.data_loader.dataset.MultiScaleDatasetVal(dataset_path, cfg, transforms=None)[source]

Class for loading aseg file with augmentations (transforms).

Methods

get_subject_names()

Get subject names.

get_subject_names()[source]

Get subject names.

class FastSurferCNN.data_loader.dataset.MultiScaleOrigDataThickSlices(orig_data, orig_zoom, cfg, transforms=None)[source]

Load MRI-Image and process it to correct format for network inference.