mix_det.yml 825 字节
Newer Older
F
Feng Ni 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
metric: COCO
num_classes: 1

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

EvalDataset:
  !COCODataSet
F
Feng Ni 已提交
14
    image_dir: images/train
F
Feng Ni 已提交
15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
    anno_path: annotations/val_half.json
    dataset_dir: dataset/mot/MOT17

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


# MOTDataset for MOT evaluation and inference
EvalMOTDataset:
  !MOTImageFolder
    dataset_dir: dataset/mot
    data_root: MOT17/images/half
    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