ssdlite_mobilenet_v1_300.yml 931 B

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. architecture: SSD
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/MobileNetV1_ssld_pretrained.pdparams
  3. SSD:
  4. backbone: MobileNet
  5. ssd_head: SSDHead
  6. post_process: BBoxPostProcess
  7. MobileNet:
  8. conv_decay: 0.00004
  9. scale: 1
  10. extra_block_filters: [[256, 512], [128, 256], [128, 256], [64, 128]]
  11. with_extra_blocks: true
  12. feature_maps: [11, 13, 14, 15, 16, 17]
  13. SSDHead:
  14. use_sepconv: True
  15. conv_decay: 0.00004
  16. anchor_generator:
  17. steps: [16, 32, 64, 100, 150, 300]
  18. aspect_ratios: [[2.], [2., 3.], [2., 3.], [2., 3.], [2., 3.], [2., 3.]]
  19. min_ratio: 20
  20. max_ratio: 95
  21. base_size: 300
  22. min_sizes: []
  23. max_sizes: []
  24. offset: 0.5
  25. flip: true
  26. clip: true
  27. min_max_aspect_ratios_order: False
  28. BBoxPostProcess:
  29. decode:
  30. name: SSDBox
  31. nms:
  32. name: MultiClassNMS
  33. keep_top_k: 200
  34. score_threshold: 0.01
  35. nms_threshold: 0.45
  36. nms_top_k: 400
  37. nms_eta: 1.0