ssdlite_mobilenet_v3_small_320.yml 1.0 KB

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