metric: COCO
num_classes: 1

# Detection Dataset for training
TrainDataset:
  !COCODataSet
    image_dir: images/train
    anno_path: annotations/train.json
    dataset_dir: dataset/mot/HT21
    data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']

EvalDataset:
  !COCODataSet
    image_dir: images/train
    anno_path: annotations/val_half.json
    dataset_dir: dataset/mot/HT21

TestDataset:
  !ImageFolder
    dataset_dir: dataset/mot/HT21
    anno_path: annotations/val_half.json


# MOTDataset for MOT evaluation and inference
EvalMOTDataset:
  !MOTImageFolder
    dataset_dir: dataset/mot
    data_root: HT21/images/test
    keep_ori_im: True # set as True in DeepSORT and ByteTrack

TestMOTDataset:
  !MOTImageFolder
    dataset_dir: dataset/mot
    keep_ori_im: True # set True if save visualization images or video