fairmot_hardnet85.yml 784 B

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. architecture: FairMOT
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/centernet_hardnet85_coco.pdparams
  3. for_mot: True
  4. FairMOT:
  5. detector: CenterNet
  6. reid: FairMOTEmbeddingHead
  7. loss: FairMOTLoss
  8. tracker: JDETracker
  9. CenterNet:
  10. backbone: HarDNet
  11. neck: CenterNetHarDNetFPN
  12. head: CenterNetHead
  13. post_process: CenterNetPostProcess
  14. HarDNet:
  15. depth_wise: False
  16. return_idx: [1,3,8,13]
  17. arch: 85
  18. CenterNetHarDNetFPN:
  19. num_layers: 85
  20. down_ratio: 4
  21. last_level: 4
  22. out_channel: 0
  23. CenterNetHead:
  24. head_planes: 128
  25. FairMOTEmbeddingHead:
  26. ch_head: 512
  27. CenterNetPostProcess:
  28. max_per_img: 500
  29. regress_ltrb: True
  30. JDETracker:
  31. conf_thres: 0.4
  32. tracked_thresh: 0.4
  33. metric_type: cosine
  34. min_box_area: 200
  35. vertical_ratio: 1.6 # for pedestrian