blazeface.yml 940 字节
Newer Older
X
xiaoting 已提交
1
architecture: BlazeFace
2

X
xiaoting 已提交
3
BlazeFace:
4
  backbone: BlazeNet
X
xiaoting 已提交
5 6
  neck: BlazeNeck
  blaze_head: FaceHead
7 8 9 10 11 12
  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]]
X
xiaoting 已提交
13 14 15 16 17
  act: relu

BlazeNeck:
  neck_type : None
  in_channel: [96,96]
18 19

FaceHead:
X
xiaoting 已提交
20
  in_channels: [96,96]
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
  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