bytetrack_yolov3.yml 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. # This config is an assembled config for ByteTrack MOT, used as eval/infer mode for MOT.
  2. _BASE_: [
  3. 'detector/yolov3_darknet53_40e_608x608_mot17half.yml',
  4. '_base_/mot17.yml',
  5. '_base_/yolov3_mot_reader_608x608.yml'
  6. ]
  7. weights: output/bytetrack_yolov3/model_final
  8. log_iter: 20
  9. snapshot_epoch: 2
  10. metric: MOT # eval/infer mode
  11. num_classes: 1
  12. architecture: ByteTrack
  13. pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/yolov3_darknet53_270e_coco.pdparams
  14. ByteTrack:
  15. detector: YOLOv3 # General YOLOv3 version
  16. reid: None
  17. tracker: JDETracker
  18. det_weights: https://bj.bcebos.com/v1/paddledet/models/mot/yolov3_darknet53_40e_608x608_mot17half.pdparams
  19. reid_weights: None
  20. YOLOv3:
  21. backbone: DarkNet
  22. neck: YOLOv3FPN
  23. yolo_head: YOLOv3Head
  24. post_process: BBoxPostProcess
  25. # Tracking requires higher quality boxes, so NMS score_threshold will be higher
  26. BBoxPostProcess:
  27. decode:
  28. name: YOLOBox
  29. conf_thresh: 0.005
  30. downsample_ratio: 32
  31. clip_bbox: true
  32. nms:
  33. name: MultiClassNMS
  34. keep_top_k: 100
  35. score_threshold: 0.01
  36. nms_threshold: 0.45
  37. nms_top_k: 1000
  38. # BYTETracker
  39. JDETracker:
  40. use_byte: True
  41. match_thres: 0.9
  42. conf_thres: 0.2
  43. low_conf_thres: 0.1
  44. min_box_area: 100
  45. vertical_ratio: 1.6 # for pedestrian