fairmot_dla34.yml 911 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. architecture: FairMOT
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/fairmot_dla34_crowdhuman_pretrained.pdparams
  3. for_mot: True
  4. FairMOT:
  5. detector: CenterNet
  6. reid: FairMOTEmbeddingHead
  7. loss: FairMOTLoss
  8. tracker: JDETracker
  9. CenterNet:
  10. backbone: DLA
  11. neck: CenterNetDLAFPN
  12. head: CenterNetHead
  13. post_process: CenterNetPostProcess
  14. CenterNetDLAFPN:
  15. down_ratio: 4
  16. last_level: 5
  17. out_channel: 0
  18. dcn_v2: True
  19. with_sge: False
  20. CenterNetHead:
  21. head_planes: 256
  22. prior_bias: -2.19
  23. regress_ltrb: True
  24. size_loss: 'L1'
  25. loss_weight: {'heatmap': 1.0, 'size': 0.1, 'offset': 1.0, 'iou': 0.0}
  26. add_iou: False
  27. FairMOTEmbeddingHead:
  28. ch_head: 256
  29. ch_emb: 128
  30. CenterNetPostProcess:
  31. max_per_img: 500
  32. down_ratio: 4
  33. regress_ltrb: True
  34. JDETracker:
  35. conf_thres: 0.4
  36. tracked_thresh: 0.4
  37. metric_type: cosine
  38. min_box_area: 200
  39. vertical_ratio: 1.6 # for pedestrian