vis4d.zoo.base.datasets.coco.detection

COCO data loading config for object detection.

Functions

get_coco_detection_cfg([data_root, ...])	Get the default config for COCO detection.
get_test_dataloader(data_root, split, ...[, ...])	Get the default test dataloader for COCO detection.
get_train_dataloader(data_root, split, ...)	Get the default train dataloader for COCO detection.

get_coco_detection_cfg(data_root='data/coco', train_split='train2017', train_keys_to_load=('images', 'boxes2d', 'boxes2d_classes'), train_cached_file_path='data/coco/train.pkl', test_split='val2017', test_keys_to_load=('images', 'original_images', 'boxes2d', 'boxes2d_classes'), test_cached_file_path='data/coco/val.pkl', cache_as_binary=True, data_backend=None, image_size=(800, 1333), samples_per_gpu=2, workers_per_gpu=2)

Get the default config for COCO detection.

Return type:

DataConfig

get_test_dataloader(data_root, split, keys_to_load, data_backend, image_size, samples_per_gpu, workers_per_gpu, cache_as_binary, cached_file_path=None)

Get the default test dataloader for COCO detection.

Return type:

ConfigDict

get_train_dataloader(data_root, split, keys_to_load, data_backend, image_size, samples_per_gpu, workers_per_gpu, cache_as_binary, cached_file_path=None)

Get the default train dataloader for COCO detection.

Return type:

ConfigDict