ssd_mobilenet_v1_300.yml 997 B

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. architecture: SSD
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ssd_mobilenet_v1_coco_pretrained.pdparams
  3. SSD:
  4. backbone: MobileNet
  5. ssd_head: SSDHead
  6. post_process: BBoxPostProcess
  7. MobileNet:
  8. norm_decay: 0.
  9. scale: 1
  10. conv_learning_rate: 0.1
  11. extra_block_filters: [[256, 512], [128, 256], [128, 256], [64, 128]]
  12. with_extra_blocks: true
  13. feature_maps: [11, 13, 14, 15, 16, 17]
  14. SSDHead:
  15. kernel_size: 1
  16. padding: 0
  17. anchor_generator:
  18. steps: [0, 0, 0, 0, 0, 0]
  19. aspect_ratios: [[2.], [2., 3.], [2., 3.], [2., 3.], [2., 3.], [2., 3.]]
  20. min_ratio: 20
  21. max_ratio: 90
  22. base_size: 300
  23. min_sizes: [60.0, 105.0, 150.0, 195.0, 240.0, 285.0]
  24. max_sizes: [[], 150.0, 195.0, 240.0, 285.0, 300.0]
  25. offset: 0.5
  26. flip: 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