centernet_mbv3_small_140e_coco.yml 579 字节
Newer Older
Z
zhoujun 已提交
1 2 3 4 5
_BASE_: [
  'centernet_r50_140e_coco.yml'
]

pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/MobileNetV3_small_x1_0_ssld_pretrained.pdparams
F
Feng Ni 已提交
6
weights: output/centernet_mbv3_small_140e_coco/model_final
Z
zhoujun 已提交
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28

CenterNet:
  backbone: MobileNetV3
  neck: CenterNetDLAFPN
  head: CenterNetHead
  post_process: CenterNetPostProcess

MobileNetV3:
  model_name: small
  scale: 1.
  with_extra_blocks: false
  extra_block_filters: []
  feature_maps: [4, 9, 12]

CenterNetDLAFPN:
  first_level: 0
  last_level: 3
  down_ratio: 8
  dcn_v2: False

TrainReader:
  batch_size: 32