mcfairmot_dla34_30e_1088x608_visdrone.yml 859 B

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. _BASE_: [
  2. '../fairmot/fairmot_dla34_30e_1088x608.yml',
  3. '../../datasets/mcmot.yml'
  4. ]
  5. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/fairmot_dla34_crowdhuman_pretrained.pdparams
  6. FairMOT:
  7. detector: CenterNet
  8. reid: FairMOTEmbeddingHead
  9. loss: FairMOTLoss
  10. tracker: JDETracker # multi-class tracker
  11. CenterNetHead:
  12. regress_ltrb: False
  13. CenterNetPostProcess:
  14. regress_ltrb: False
  15. max_per_img: 200
  16. JDETracker:
  17. min_box_area: 0
  18. vertical_ratio: 0 # no need to filter bboxes according to w/h
  19. conf_thres: 0.4
  20. tracked_thresh: 0.4
  21. metric_type: cosine
  22. weights: output/mcfairmot_dla34_30e_1088x608_visdrone/model_final
  23. epoch: 30
  24. LearningRate:
  25. base_lr: 0.0005
  26. schedulers:
  27. - !PiecewiseDecay
  28. gamma: 0.1
  29. milestones: [10, 20]
  30. use_warmup: False
  31. OptimizerBuilder:
  32. optimizer:
  33. type: Adam
  34. regularizer: NULL