picodet_s_320_coco.yml 751 字节
Newer Older
G
Guanghua Yu 已提交
1 2 3
_BASE_: [
  '../datasets/coco_detection.yml',
  '../runtime.yml',
4
  '_base_/picodet_esnet.yml',
G
Guanghua Yu 已提交
5
  '_base_/optimizer_300e.yml',
G
Guanghua Yu 已提交
6 7 8
  '_base_/picodet_320_reader.yml',
]

9 10
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ESNet_x0_75_pretrained.pdparams
weights: output/picodet_s_320_coco/model_final
G
Guanghua Yu 已提交
11 12
find_unused_parameters: True
use_ema: true
G
Guanghua Yu 已提交
13
cycle_epoch: 40
G
Guanghua Yu 已提交
14 15
snapshot_epoch: 10

16 17 18 19 20
ESNet:
  scale: 0.75
  feature_maps: [4, 11, 14]
  act: hard_swish
  channel_ratio: [0.875, 0.5, 0.5, 0.5, 0.625, 0.5, 0.625, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5]
G
Guanghua Yu 已提交
21

22 23
CSPPAN:
  out_channels: 96
G
Guanghua Yu 已提交
24 25 26 27

PicoHead:
  conv_feat:
    name: PicoFeat
28 29
    feat_in: 96
    feat_out: 96
G
Guanghua Yu 已提交
30
    num_convs: 2
31
    num_fpn_stride: 4
G
Guanghua Yu 已提交
32 33
    norm_type: bn
    share_cls_reg: True
34
  feat_in_chan: 96