blazeface_fpn.yml 1.0 KB
Newer Older
X
xiaoting 已提交
1
architecture: BlazeFace
2
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/blazenet_pretrain.pdparams
X
xiaoting 已提交
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46

BlazeFace:
  backbone: BlazeNet
  neck: BlazeNeck
  blaze_head: FaceHead
  post_process: BBoxPostProcess

BlazeNet:
  blaze_filters: [[24, 24], [24, 24], [24, 48, 2], [48, 48], [48, 48]]
  double_blaze_filters: [[48, 24, 96, 2], [96, 24, 96], [96, 24, 96],
                          [96, 24, 96, 2], [96, 24, 96], [96, 24, 96]]
  act: hard_swish

BlazeNeck:
  neck_type : fpn_ssh
  in_channel: [96,96]

FaceHead:
  in_channels: [48, 48]
  anchor_generator: AnchorGeneratorSSD
  loss: SSDLoss

SSDLoss:
  overlap_threshold: 0.35

AnchorGeneratorSSD:
  steps: [8., 16.]
  aspect_ratios: [[1.], [1.]]
  min_sizes: [[16.,24.], [32., 48., 64., 80., 96., 128.]]
  max_sizes: [[], []]
  offset: 0.5
  flip: False
  min_max_aspect_ratios_order: false

BBoxPostProcess:
  decode:
    name: SSDBox
  nms:
    name: MultiClassNMS
    keep_top_k: 750
    score_threshold: 0.01
    nms_threshold: 0.3
    nms_top_k: 5000
    nms_eta: 1.0