fairmot_dla34_30e_1088x608_visdrone_pedestrian.yml 781 B

1234567891011121314151617181920212223242526
  1. _BASE_: [
  2. '../fairmot/fairmot_dla34_30e_1088x608.yml'
  3. ]
  4. weights: output/fairmot_dla34_30e_1088x608_visdrone_pedestrian/model_final
  5. # for MOT training
  6. TrainDataset:
  7. !MOTDataSet
  8. dataset_dir: dataset/mot
  9. image_lists: ['visdrone_pedestrian.train']
  10. data_fields: ['image', 'gt_bbox', 'gt_class', 'gt_ide']
  11. # for MOT evaluation
  12. # If you want to change the MOT evaluation dataset, please modify 'data_root'
  13. EvalMOTDataset:
  14. !MOTImageFolder
  15. dataset_dir: dataset/mot
  16. data_root: visdrone_pedestrian/images/val
  17. keep_ori_im: False # set True if save visualization images or video, or used in DeepSORT
  18. # for MOT video inference
  19. TestMOTDataset:
  20. !MOTImageFolder
  21. dataset_dir: dataset/mot
  22. keep_ori_im: True # set True if save visualization images or video