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

G
Guanghua Yu 已提交
9 10
pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/PPLCNet_x2_0_pretrained.pdparams
weights: output/picodet_l_320_coco/best_model
G
Guanghua Yu 已提交
11 12
find_unused_parameters: True
use_ema: true
G
Guanghua Yu 已提交
13
epoch: 250
G
Guanghua Yu 已提交
14
snapshot_epoch: 10
G
Guanghua Yu 已提交
15

G
Guanghua Yu 已提交
16 17 18
LCNet:
  scale: 2.0
  feature_maps: [3, 4, 5]
G
Guanghua Yu 已提交
19

G
Guanghua Yu 已提交
20
LCPAN:
G
Guanghua Yu 已提交
21 22
  out_channels: 160

G
Guanghua Yu 已提交
23
PicoHeadV2:
G
Guanghua Yu 已提交
24 25
  conv_feat:
    name: PicoFeat
G
Guanghua Yu 已提交
26 27
    feat_in: 160
    feat_out: 160
28 29
    num_convs: 4
    num_fpn_stride: 4
G
Guanghua Yu 已提交
30
    norm_type: bn
G
Guanghua Yu 已提交
31
    share_cls_reg: True
G
Guanghua Yu 已提交
32
    use_se: True
G
Guanghua Yu 已提交
33
  feat_in_chan: 160
G
Guanghua Yu 已提交
34 35

LearningRate:
G
Guanghua Yu 已提交
36
  base_lr: 0.12
G
Guanghua Yu 已提交
37 38
  schedulers:
  - !CosineDecay
39
    max_epochs: 300
G
Guanghua Yu 已提交
40 41 42
  - !LinearWarmup
    start_factor: 0.1
    steps: 300
G
Guanghua Yu 已提交
43 44 45

TrainReader:
  batch_size: 24