jde_darknet53_30e_576x320.yml 1007 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. _BASE_: [
  2. '../../datasets/mot.yml',
  3. '../../runtime.yml',
  4. '_base_/optimizer_30e.yml',
  5. '_base_/jde_darknet53.yml',
  6. '_base_/jde_reader_576x320.yml',
  7. ]
  8. weights: output/jde_darknet53_30e_576x320/model_final
  9. JDE:
  10. detector: YOLOv3
  11. reid: JDEEmbeddingHead
  12. tracker: JDETracker
  13. YOLOv3:
  14. backbone: DarkNet
  15. neck: YOLOv3FPN
  16. yolo_head: YOLOv3Head
  17. post_process: JDEBBoxPostProcess
  18. for_mot: True
  19. YOLOv3Head:
  20. anchors: [[85,255], [120,360], [170,420], [340,420],
  21. [21,64], [30,90], [43,128], [60,180],
  22. [6,16], [8,23], [11,32], [16,45]]
  23. anchor_masks: [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]]
  24. loss: JDEDetectionLoss
  25. JDETracker:
  26. det_thresh: 0.3
  27. track_buffer: 30
  28. min_box_area: 200
  29. motion: KalmanFilter
  30. JDEBBoxPostProcess:
  31. decode:
  32. name: JDEBox
  33. conf_thresh: 0.5
  34. downsample_ratio: 32
  35. nms:
  36. name: MultiClassNMS
  37. keep_top_k: 500
  38. score_threshold: 0.01
  39. nms_threshold: 0.4
  40. nms_top_k: 2000
  41. normalized: true
  42. return_index: true