vis4d.zoo.base.datasets.imagenet

ImageNet classification config.

Functions

get_imagenet_cls_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 ImageNet 1K dataset.

get_imagenet_cls_cfg(data_root='data/imagenet', train_split='train', train_keys_to_load=('images', 'categories'), test_split='val', test_keys_to_load=('images', 'categories'), image_size=(224, 224), samples_per_gpu=256, workers_per_gpu=8)

Get the default config for COCO detection.

Return type:

DataConfig

get_test_dataloader(data_root, split, keys_to_load, image_size, samples_per_gpu, workers_per_gpu, crop_pct=0.875)

Get the default test dataloader for COCO detection.

Return type:

ConfigDict

get_train_dataloader(data_root, split, keys_to_load, image_size, samples_per_gpu, workers_per_gpu)

Get the default train dataloader for ImageNet 1K dataset.

Return type:

ConfigDict