yolov3_darknet53_40e_608x608_mot17half.yml 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. # This config is an assembled config for ByteTrack MOT, used as eval/infer mode for MOT.
  2. _BASE_: [
  3. '../../../yolov3/yolov3_darknet53_270e_coco.yml',
  4. '../_base_/mot17.yml',
  5. ]
  6. weights: output/yolov3_darknet53_40e_608x608_mot17half/model_final
  7. log_iter: 20
  8. snapshot_epoch: 2
  9. # schedule configuration for fine-tuning
  10. epoch: 40
  11. LearningRate:
  12. base_lr: 0.0001
  13. schedulers:
  14. - !PiecewiseDecay
  15. gamma: 0.1
  16. milestones:
  17. - 32
  18. - 36
  19. - !LinearWarmup
  20. start_factor: 0.3333333333333333
  21. steps: 100
  22. OptimizerBuilder:
  23. optimizer:
  24. momentum: 0.9
  25. type: Momentum
  26. regularizer:
  27. factor: 0.0005
  28. type: L2
  29. TrainReader:
  30. batch_size: 8
  31. mixup_epoch: 35
  32. # detector configuration
  33. architecture: YOLOv3
  34. pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/yolov3_darknet53_270e_coco.pdparams
  35. norm_type: sync_bn
  36. YOLOv3:
  37. backbone: DarkNet
  38. neck: YOLOv3FPN
  39. yolo_head: YOLOv3Head
  40. post_process: BBoxPostProcess
  41. DarkNet:
  42. depth: 53
  43. return_idx: [2, 3, 4]
  44. # use default config
  45. # YOLOv3FPN:
  46. YOLOv3Head:
  47. anchors: [[10, 13], [16, 30], [33, 23],
  48. [30, 61], [62, 45], [59, 119],
  49. [116, 90], [156, 198], [373, 326]]
  50. anchor_masks: [[6, 7, 8], [3, 4, 5], [0, 1, 2]]
  51. loss: YOLOv3Loss
  52. YOLOv3Loss:
  53. ignore_thresh: 0.7
  54. downsample: [32, 16, 8]
  55. label_smooth: false
  56. BBoxPostProcess:
  57. decode:
  58. name: YOLOBox
  59. conf_thresh: 0.005
  60. downsample_ratio: 32
  61. clip_bbox: true
  62. nms:
  63. name: MultiClassNMS
  64. keep_top_k: 100
  65. score_threshold: 0.01
  66. nms_threshold: 0.45
  67. nms_top_k: 1000