fairmot_dla34_30e_1088x608_bytetracker.yml 936 B

12345678910111213141516171819202122232425262728293031
  1. _BASE_: [
  2. '../../datasets/mot.yml',
  3. '../../runtime.yml',
  4. '_base_/optimizer_30e.yml',
  5. '_base_/fairmot_dla34.yml',
  6. '_base_/fairmot_reader_1088x608.yml',
  7. ]
  8. weights: output/fairmot_dla34_30e_1088x608_bytetracker/model_final
  9. # for ablation study, MIX + MOT17-half
  10. TrainDataset:
  11. !MOTDataSet
  12. dataset_dir: dataset/mot
  13. image_lists: ['mot17.half', 'caltech.all', 'cuhksysu.train', 'prw.train', 'citypersons.train', 'eth.train']
  14. data_fields: ['image', 'gt_bbox', 'gt_class', 'gt_ide']
  15. # for MOT evaluation
  16. # If you want to change the MOT evaluation dataset, please modify 'data_root'
  17. EvalMOTDataset:
  18. !MOTImageFolder
  19. dataset_dir: dataset/mot
  20. data_root: MOT17/images/half
  21. keep_ori_im: False # set True if save visualization images or video, or used in DeepSORT
  22. JDETracker:
  23. use_byte: True
  24. match_thres: 0.8
  25. conf_thres: 0.4
  26. low_conf_thres: 0.2
  27. min_box_area: 200
  28. vertical_ratio: 1.6 # for pedestrian