jde_darknet53.yml 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. architecture: JDE
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/DarkNet53_pretrained.pdparams
  3. find_unused_parameters: True
  4. JDE:
  5. detector: YOLOv3
  6. reid: JDEEmbeddingHead
  7. tracker: JDETracker
  8. YOLOv3:
  9. backbone: DarkNet
  10. neck: YOLOv3FPN
  11. yolo_head: YOLOv3Head
  12. post_process: JDEBBoxPostProcess
  13. for_mot: True
  14. DarkNet:
  15. depth: 53
  16. return_idx: [2, 3, 4]
  17. freeze_norm: True
  18. YOLOv3FPN:
  19. freeze_norm: True
  20. YOLOv3Head:
  21. anchors: [[128,384], [180,540], [256,640], [512,640],
  22. [32,96], [45,135], [64,192], [90,271],
  23. [8,24], [11,34], [16,48], [23,68]]
  24. anchor_masks: [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]]
  25. loss: JDEDetectionLoss
  26. JDEBBoxPostProcess:
  27. decode:
  28. name: JDEBox
  29. conf_thresh: 0.3
  30. downsample_ratio: 32
  31. nms:
  32. name: MultiClassNMS
  33. keep_top_k: 500
  34. score_threshold: 0.01
  35. nms_threshold: 0.5
  36. nms_top_k: 2000
  37. normalized: true
  38. JDEEmbeddingHead:
  39. anchor_levels: 3
  40. anchor_scales: 4
  41. embedding_dim: 512
  42. emb_loss: JDEEmbeddingLoss
  43. jde_loss: JDELoss
  44. JDETracker:
  45. det_thresh: 0.3
  46. track_buffer: 30
  47. min_box_area: 200
  48. vertical_ratio: 1.6 # for pedestrian