mot.yml 728 B

1234567891011121314151617181920212223
  1. metric: MOT
  2. num_classes: 1
  3. # for MOT training
  4. TrainDataset:
  5. !MOTDataSet
  6. dataset_dir: dataset/mot
  7. image_lists: ['mot17.train', 'caltech.all', 'cuhksysu.train', 'prw.train', 'citypersons.train', 'eth.train']
  8. data_fields: ['image', 'gt_bbox', 'gt_class', 'gt_ide']
  9. # for MOT evaluation
  10. # If you want to change the MOT evaluation dataset, please modify 'data_root'
  11. EvalMOTDataset:
  12. !MOTImageFolder
  13. dataset_dir: dataset/mot
  14. data_root: MOT16/images/train
  15. keep_ori_im: False # set True if save visualization images or video, or used in DeepSORT
  16. # for MOT video inference
  17. TestMOTDataset:
  18. !MOTImageFolder
  19. dataset_dir: dataset/mot
  20. keep_ori_im: True # set True if save visualization images or video