composer.datasets.brats_hparams#

BraTS (Brain Tumor Segmentation) dataset hyperparameters.

Functions

dataclass

Returns the same class as was passed in, with dunder methods added based on the fields defined in the class.

get_data_split

composer.datasets.brats.get_data_split

Classes

PytTrain

composer.datasets.brats.PytTrain

PytVal

composer.datasets.brats.PytVal

Hparams

These classes are used with yahp for YAML-based configuration.

BratsDatasetHparams

Defines an instance of the BraTS dataset for image segmentation.

DataLoaderHparams

Hyperparameters to initialize a torch.utils.data.DataLoader.

DatasetHparams

Abstract base class for hyperparameters to initialize a dataset.

Attributes

  • Optional

class composer.datasets.brats_hparams.BratsDatasetHparams(drop_last=True, shuffle=True, oversampling=0.33, is_train=True, datadir=None)[source]#

Bases: composer.datasets.dataset_hparams.DatasetHparams

Defines an instance of the BraTS dataset for image segmentation.

Parameters

oversampling (float) โ€“ The oversampling ratio to use. Default: 0.33.