Use Builtin Datasets
zcls implements the following datasets:
CIFAR10
Modify the configuration file as follows:
DATASET:
NAME: 'CIFAR10'
TRAIN_ROOT: /path/to/train_root
TEST_ROOT: /path/to/test_root
TOP_K: (1, 5)
Note: If the path is not set correctly or the file does not exist, zcls will automatically download the dataset
CIFAR100
Modify the configuration file as follows:
DATASET:
NAME: 'CIFAR100'
TRAIN_ROOT: /path/to/train_root
TEST_ROOT: /path/to/test_root
TOP_K: (1, 5)
Note: If the path is not set correctly or the file does not exist, zcls will automatically download the dataset
FashionMNIST
Modify the configuration file as follows:
DATASET:
NAME: 'FashionMNIST'
TRAIN_ROOT: /path/to/train_root
TEST_ROOT: /path/to/test_root
TOP_K: (1, 5)
Note: If the path is not set correctly or the file does not exist, zcls will automatically download the dataset
ImageNet
Modify the configuration file as follows:
DATASET:
NAME: 'ImageNet'
TRAIN_ROOT: /path/to/train_root
TEST_ROOT: /path/to/test_root
TOP_K: (1, 5)
Download the following datasets to the data directory
.
├── ILSVRC2012_devkit_t12.tar.gz
├── ILSVRC2012_devkit_t3.tar.gz
├── ILSVRC2012_img_test.tar
├── ILSVRC2012_img_train_t3.tar
├── ILSVRC2012_img_train.tar
├── ILSVRC2012_img_val.tar