fairmot_hrnetv2_w18_dlafpn.yml 708 B

1234567891011121314151617181920212223242526272829303132333435363738
  1. architecture: FairMOT
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/HRNet_W18_C_pretrained.pdparams
  3. for_mot: True
  4. FairMOT:
  5. detector: CenterNet
  6. reid: FairMOTEmbeddingHead
  7. loss: FairMOTLoss
  8. tracker: JDETracker
  9. CenterNet:
  10. backbone: HRNet
  11. head: CenterNetHead
  12. post_process: CenterNetPostProcess
  13. neck: CenterNetDLAFPN
  14. HRNet:
  15. width: 18
  16. freeze_at: 0
  17. return_idx: [0, 1, 2, 3]
  18. upsample: False
  19. CenterNetDLAFPN:
  20. down_ratio: 4
  21. last_level: 3
  22. out_channel: 0
  23. first_level: 0
  24. dcn_v2: False
  25. CenterNetPostProcess:
  26. max_per_img: 500
  27. JDETracker:
  28. conf_thres: 0.4
  29. tracked_thresh: 0.4
  30. metric_type: cosine
  31. min_box_area: 200
  32. vertical_ratio: 1.6 # for pedestrian