ht21.yml 829 B

12345678910111213141516171819202122232425262728293031323334
  1. metric: COCO
  2. num_classes: 1
  3. # Detection Dataset for training
  4. TrainDataset:
  5. !COCODataSet
  6. image_dir: images/train
  7. anno_path: annotations/train.json
  8. dataset_dir: dataset/mot/HT21
  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/HT21
  15. TestDataset:
  16. !ImageFolder
  17. dataset_dir: dataset/mot/HT21
  18. anno_path: annotations/val_half.json
  19. # MOTDataset for MOT evaluation and inference
  20. EvalMOTDataset:
  21. !MOTImageFolder
  22. dataset_dir: dataset/mot
  23. data_root: HT21/images/test
  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