vis4d.zoo.base.datasets.coco.sem_seg

COCO data loading config for for semantic segmentation.

Functions

get_coco_sem_seg_cfg([data_root, ...])	Get the default config for COCO semantic segmentation.
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_sem_seg_cfg(data_root='data/coco', train_split='train2017', train_keys_to_load=('images', 'seg_masks'), test_split='val2017', test_keys_to_load=('images', 'seg_masks'), data_backend=None, image_size=(520, 520), samples_per_gpu=2, workers_per_gpu=2)

Get the default config for COCO semantic segmentation.

Return type:

DataConfig

get_test_dataloader(data_root, split, keys_to_load, data_backend, image_size, samples_per_gpu, workers_per_gpu)

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)

Get the default train dataloader for COCO detection.

Return type:

ConfigDict