BratsDatasetHparams

class composer.datasets.BratsDatasetHparams(is_train: bool, datadir: str, download: bool, drop_last: bool = True, shuffle: bool = True, oversampling: float = 0.33)

Bases: composer.datasets.hparams.DatasetHparams

Defines an instance of the BraTS dataset for image segmentation.

Parameters

• is_train (bool) – Whether to load the training or validation dataset.

• datadir (str) – Data directory to use.

• download (bool) – Whether to download the dataset, if needed.

• drop_last (bool) – Whether to drop the last samples for the last batch.

• shuffle (bool) – Whether to shuffle the dataset for each epoch.

• oversampling (float) – The oversampling ratio to use.

initialize_object() → DataloaderSpec

Initializes a DataloaderSpec for this dataset.