mix_mot_ch.yml 828 B

12345678910111213141516171819202122232425262728293031323334
  1. metric: COCO
  2. num_classes: 1
  3. # Detection Dataset for training
  4. TrainDataset:
  5. !COCODataSet
  6. image_dir: ""
  7. anno_path: annotations/train.json
  8. dataset_dir: dataset/mot/mix_mot_ch
  9. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  10. EvalDataset:
  11. !COCODataSet
  12. image_dir: images/train
  13. anno_path: annotations/val_half.json
  14. dataset_dir: dataset/mot/MOT17
  15. TestDataset:
  16. !ImageFolder
  17. anno_path: annotations/val_half.json
  18. dataset_dir: dataset/mot/MOT17
  19. # MOTDataset for MOT evaluation and inference
  20. EvalMOTDataset:
  21. !MOTImageFolder
  22. dataset_dir: dataset/mot
  23. data_root: MOT17/images/half
  24. keep_ori_im: True # set as True in DeepSORT and ByteTrack
  25. TestMOTDataset:
  26. !MOTImageFolder
  27. dataset_dir: dataset/mot
  28. keep_ori_im: True # set True if save visualization images or video