dgs.models.dataset.dataset

Module for handling data loading and preprocessing using torch Datasets.

Module Classes

BBoxDataset(*args, **kwargs)	A dataset using the bounding boxes as indices.
BaseDataset(*args, **kwargs)	Base class for custom datasets.
ImageDataset(*args, **kwargs)	A dataset using the image IDs as indices.
ImageHistoryDataset(*args, **kwargs)	A dataset with one index per image ID, the main difference is that in addition to the current frame, the last L frames are given as well.
VideoDataset(*args, **kwargs)	A dataset containing a single video.