vis4d.zoo.yolox.data

COCO data loading config for YOLOX object detection.

Functions

get_coco_yolox_cfg([data_root, train_split, ...])	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_yolox_cfg(data_root='data/coco', train_split='train2017', train_keys_to_load=('images', 'boxes2d', 'boxes2d_classes'), test_split='val2017', test_keys_to_load=('images', 'original_images'), data_backend=None, train_image_size=(640, 640), scaling_ratio_range=(0.1, 2.0), use_mixup=True, test_image_size=(640, 640), 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)

Get the default test dataloader for COCO detection.

Return type:

ConfigDict

get_train_dataloader(data_root, split, keys_to_load, data_backend, image_size, scaling_ratio_range, use_mixup, samples_per_gpu, workers_per_gpu)

Get the default train dataloader for COCO detection.

Return type:

ConfigDict