FastSurferCNN.data_loader.loader¶

FastSurferCNN.data_loader.loader.get_dataloader(cfg, mode)[source]¶

Create the dataset and pytorch data loader.

Parameters:

cfgyacs.config.CfgNode: Configuration node.
modestr: Loading data for train, val and test mode.

Returns:

torch.utils.data.DataLoader: Dataloader with given configs and mode.