fcos_r50_fpn.yml 916 字节
Newer Older
F
Feng Ni 已提交
1
architecture: FCOS
2
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_cos_pretrained.pdparams
F
Feng Ni 已提交
3 4 5 6 7 8 9 10

FCOS:
  backbone: ResNet
  neck: FPN
  fcos_head: FCOSHead

ResNet:
  depth: 50
F
Feng Ni 已提交
11
  variant: 'b'
F
Feng Ni 已提交
12
  norm_type: bn
F
Feng Ni 已提交
13 14
  freeze_at: 0 # res2
  return_idx: [1, 2, 3]
F
Feng Ni 已提交
15 16 17 18
  num_stages: 4

FPN:
  out_channel: 256
F
Feng Ni 已提交
19 20
  spatial_scales: [0.125, 0.0625, 0.03125]
  extra_stage: 2
F
Feng Ni 已提交
21 22
  has_extra_convs: True
  use_c5: False
F
Feng Ni 已提交
23 24 25 26 27 28 29 30

FCOSHead:
  fcos_feat:
    name: FCOSFeat
    feat_in: 256
    feat_out: 256
    num_convs: 4
    norm_type: "gn"
F
Feng Ni 已提交
31
    use_dcn: False
F
Feng Ni 已提交
32 33
  fpn_stride: [8, 16, 32, 64, 128]
  prior_prob: 0.01
F
Feng Ni 已提交
34 35 36 37 38 39 40 41 42
  norm_reg_targets: True
  centerness_on_reg: True
  num_shift: 0.5
  fcos_loss:
    name: FCOSLoss
    loss_alpha: 0.25
    loss_gamma: 2.0
    iou_loss_type: "giou"
    reg_weights: 1.0
F
Feng Ni 已提交
43 44 45 46 47 48
  nms:
    name: MultiClassNMS
    nms_top_k: 1000
    keep_top_k: 100
    score_threshold: 0.025
    nms_threshold: 0.6