ssd_r34_300.yml 851 B

1234567891011121314151617181920212223242526272829303132333435363738
  1. architecture: SSD
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet34_pretrained.pdparams
  3. SSD:
  4. backbone: ResNet
  5. ssd_head: SSDHead
  6. post_process: BBoxPostProcess
  7. r34_backbone: True
  8. ResNet:
  9. # index 0 stands for res2
  10. depth: 34
  11. norm_type: bn
  12. freeze_norm: False
  13. freeze_at: -1
  14. return_idx: [2]
  15. num_stages: 3
  16. SSDHead:
  17. anchor_generator:
  18. steps: [8, 16, 32, 64, 100, 300]
  19. aspect_ratios: [[2.], [2., 3.], [2., 3.], [2., 3.], [2.], [2.]]
  20. min_sizes: [21.0, 45.0, 99.0, 153.0, 207.0, 261.0]
  21. max_sizes: [45.0, 99.0, 153.0, 207.0, 261.0, 315.0]
  22. offset: 0.5
  23. clip: True
  24. min_max_aspect_ratios_order: True
  25. use_extra_head: True
  26. BBoxPostProcess:
  27. decode:
  28. name: SSDBox
  29. nms:
  30. name: MultiClassNMS
  31. keep_top_k: 200
  32. score_threshold: 0.05
  33. nms_threshold: 0.5
  34. nms_top_k: 400