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 the subject name.
unify_imgs
(img, label, weight)Pad img, label and weight.
- unify_imgs(img, label, weight)[source]¶
Pad img, label and weight.
- Parameters:
- img
npt.NDArray
Image to unify.
- label
npt.NDArray
Labels of the image.
- weight
npt.NDArray
Weights of the image.
- img
- Returns:
np.ndarray
Img.
np.ndarray
Label.
np.ndarray
Weight.