GeneralDataset

ZCls provides two custom dataset classes for common file storage paths.

GeneralDatasetV1

Suppose your dataset is in the following format

root/dog/xxx.png
root/dog/xxy.png
root/dog/xxz.png

root/cat/123.png
root/cat/nsdf3.png
root/cat/asd932_.png

modify config_file like this

DATASET:
  NAME: 'GeneralDataset'
  TRAIN_ROOT: /path/to/train_root
  TEST_ROOT: /path/to/test_root
  TOP_K: (1, 5)

Suppose your dataset is in the following format

{
  'class1': [img1, img2, ...],
  'class2': [img1, img2, ...],
  ...
}

modify config_file like this

DATASET:
  NAME: 'GeneralDatasetV2'
  TRAIN_ROOT: /path/to/train_path
  TEST_ROOT: /path/to/test_path
  TOP_K: (1, 5)