blazeface_fpn.yml 950 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. architecture: BlazeFace
  2. BlazeFace:
  3. backbone: BlazeNet
  4. neck: BlazeNeck
  5. blaze_head: FaceHead
  6. post_process: BBoxPostProcess
  7. BlazeNet:
  8. blaze_filters: [[24, 24], [24, 24], [24, 48, 2], [48, 48], [48, 48]]
  9. double_blaze_filters: [[48, 24, 96, 2], [96, 24, 96], [96, 24, 96],
  10. [96, 24, 96, 2], [96, 24, 96], [96, 24, 96]]
  11. act: hard_swish
  12. BlazeNeck:
  13. neck_type : fpn_ssh
  14. in_channel: [96,96]
  15. FaceHead:
  16. in_channels: [48, 48]
  17. anchor_generator: AnchorGeneratorSSD
  18. loss: SSDLoss
  19. SSDLoss:
  20. overlap_threshold: 0.35
  21. AnchorGeneratorSSD:
  22. steps: [8., 16.]
  23. aspect_ratios: [[1.], [1.]]
  24. min_sizes: [[16.,24.], [32., 48., 64., 80., 96., 128.]]
  25. max_sizes: [[], []]
  26. offset: 0.5
  27. flip: False
  28. min_max_aspect_ratios_order: false
  29. BBoxPostProcess:
  30. decode:
  31. name: SSDBox
  32. nms:
  33. name: MultiClassNMS
  34. keep_top_k: 750
  35. score_threshold: 0.01
  36. nms_threshold: 0.3
  37. nms_top_k: 5000
  38. nms_eta: 1.0