vis4d.data.datasets.shift

SHIFT dataset.

Classes

SHIFT(data_root, split[, keys_to_load, ...])

SHIFT dataset class, supporting multiple tasks and views.

class SHIFT(data_root, split, keys_to_load=('images', 'boxes2d'), views_to_load=('front', ), attributes_to_load=None, framerate='images', shift_type='discrete', skip_empty_frames=False, backend=<vis4d.data.io.hdf5.HDF5Backend object>, num_workers=1, verbose=False)

SHIFT dataset class, supporting multiple tasks and views.

Initialize SHIFT dataset.

__getitem__(idx)

Get single sample.

Parameters:

idx (int) – Index of sample.

Returns:

sample at index in Vis4D input format.

Return type:

DictData

__len__()

Get the number of samples in the dataset.

Return type:

int

validate_keys(keys_to_load)

Validate that all keys to load are supported.

Return type:

None