vis4d.zoo.cc_3dt.data

CC-3DT NuScenes data config.

Functions

get_nusc_cfg([data_root, version, ...])

Get the default config for nuScenes tracking.

get_test_dataloader(test_dataset, ...)

Get the default test dataloader for nuScenes tracking.

get_train_dataloader(train_dataset, ...)

Get the default train dataloader for nuScenes tracking.

get_nusc_cfg(data_root='data/nuscenes', version='v1.0-trainval', train_split='train', test_split='val', data_backend=None, samples_per_gpu=2, workers_per_gpu=2)[source]

Get the default config for nuScenes tracking.

Return type:

DataConfig

get_test_dataloader(test_dataset, samples_per_gpu, workers_per_gpu)[source]

Get the default test dataloader for nuScenes tracking.

Return type:

ConfigDict

get_train_dataloader(train_dataset, samples_per_gpu, workers_per_gpu)[source]

Get the default train dataloader for nuScenes tracking.

Return type:

ConfigDict