centertrack_dla34.yml 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/crowdhuman_centertrack.pdparams
  2. architecture: CenterTrack
  3. for_mot: True
  4. mot_metric: True
  5. ### model
  6. CenterTrack:
  7. detector: CenterNet
  8. plugin_head: CenterTrackHead
  9. tracker: CenterTracker
  10. ### CenterTrack.detector
  11. CenterNet:
  12. backbone: DLA
  13. neck: CenterNetDLAFPN
  14. head: CenterNetHead
  15. post_process: CenterNetPostProcess
  16. for_mot: True # Note
  17. DLA:
  18. depth: 34
  19. pre_img: True # Note
  20. pre_hm: True # Note
  21. CenterNetDLAFPN:
  22. down_ratio: 4
  23. last_level: 5
  24. out_channel: 0
  25. dcn_v2: True
  26. CenterNetHead:
  27. head_planes: 256
  28. prior_bias: -4.6 # Note
  29. regress_ltrb: False
  30. size_loss: 'L1'
  31. loss_weight: {'heatmap': 1.0, 'size': 0.1, 'offset': 1.0}
  32. CenterNetPostProcess:
  33. max_per_img: 100 # top-K
  34. regress_ltrb: False
  35. ### CenterTrack.plugin_head
  36. CenterTrackHead:
  37. head_planes: 256
  38. task: tracking
  39. loss_weight: {'tracking': 1.0, 'ltrb_amodal': 0.1}
  40. add_ltrb_amodal: True
  41. ### CenterTrack.tracker
  42. CenterTracker:
  43. min_box_area: -1
  44. vertical_ratio: -1
  45. track_thresh: 0.4
  46. pre_thresh: 0.5