diffusion_models.spine_dataset.base_dataset.BaseDataset

class diffusion_models.spine_dataset.base_dataset.BaseDataset(resolution, random_crop, crop_size, mode, **kwargs)

Bases: Dataset, ABC

Interface for Datasets in Spine Diffusion package.

This interface is currently not enforced, but any dataset implementation should follow the guidelines outlined here, this is especially true for the exact returns of the __getitem__ method.

__init__(resolution, random_crop, crop_size, mode, **kwargs)

Constructor of BaseDataset.

Parameters:

• resolution (int) – determines base resolution of the dataset, i.e. a dataset with an original size of 256 (in 3D) will be downsampled to that resolution

• random_crop (bool)

Methods

__init__(resolution, random_crop, crop_size, ...)	Constructor of BaseDataset.
check_output(output)