fairmot_dla34_30e_1088x608_airplane.yml 949 B

123456789101112131415161718192021222324252627282930313233
  1. _BASE_: [
  2. 'fairmot_dla34_30e_1088x608.yml',
  3. ]
  4. pretrain_weights: https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_1088x608.pdparams
  5. weights: output/fairmot_dla34_30e_1088x608_airplane/model_final
  6. JDETracker:
  7. conf_thres: 0.4
  8. tracked_thresh: 0.4
  9. metric_type: cosine
  10. min_box_area: 0
  11. vertical_ratio: 0
  12. # for MOT training
  13. TrainDataset:
  14. !MOTDataSet
  15. dataset_dir: dataset/mot
  16. image_lists: ['airplane.train']
  17. data_fields: ['image', 'gt_bbox', 'gt_class', 'gt_ide']
  18. # for MOT evaluation
  19. # If you want to change the MOT evaluation dataset, please modify 'data_root'
  20. EvalMOTDataset:
  21. !MOTImageFolder
  22. dataset_dir: dataset/mot
  23. data_root: airplane/images/train
  24. keep_ori_im: False # set True if save visualization images or video, or used in DeepSORT
  25. # for MOT video inference
  26. TestMOTDataset:
  27. !MOTImageFolder
  28. dataset_dir: dataset/mot
  29. keep_ori_im: True # set True if save visualization images or video