mcmot.yml 803 B

12345678910111213141516171819202122232425
  1. metric: MCMOT
  2. num_classes: 10
  3. # using VisDrone2019 MOT dataset with 10 classes as default, you can modify it for your needs.
  4. # for MCMOT training
  5. TrainDataset:
  6. !MCMOTDataSet
  7. dataset_dir: dataset/mot
  8. image_lists: ['visdrone_mcmot.train']
  9. data_fields: ['image', 'gt_bbox', 'gt_class', 'gt_ide']
  10. label_list: label_list.txt
  11. # for MCMOT evaluation
  12. # If you want to change the MCMOT evaluation dataset, please modify 'data_root'
  13. EvalMOTDataset:
  14. !MOTImageFolder
  15. dataset_dir: dataset/mot
  16. data_root: visdrone_mcmot/images/val
  17. keep_ori_im: False # set True if save visualization images or video, or used in DeepSORT
  18. # for MCMOT video inference
  19. TestMOTDataset:
  20. !MOTImageFolder
  21. dataset_dir: dataset/mot
  22. keep_ori_im: True # set True if save visualization images or video