fcos_r50_fpn.yml 1017 字节
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 11 12 13 14 15 16 17 18 19

FCOS:
  backbone: ResNet
  neck: FPN
  fcos_head: FCOSHead
  fcos_post_process: FCOSPostProcess

ResNet:
  # index 0 stands for res2
  depth: 50
  norm_type: bn
  freeze_at: 0
  return_idx: [1,2,3]
  num_stages: 4

FPN:
  out_channel: 256
F
Feng Ni 已提交
20 21
  spatial_scales: [0.125, 0.0625, 0.03125]
  extra_stage: 2
F
Feng Ni 已提交
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55
  has_extra_convs: true
  use_c5: false

FCOSHead:
  fcos_feat:
    name: FCOSFeat
    feat_in: 256
    feat_out: 256
    num_convs: 4
    norm_type: "gn"
    use_dcn: false
  num_classes: 80
  fpn_stride: [8, 16, 32, 64, 128]
  prior_prob: 0.01
  fcos_loss: FCOSLoss
  norm_reg_targets: true
  centerness_on_reg: true

FCOSLoss:
  loss_alpha: 0.25
  loss_gamma: 2.0
  iou_loss_type: "giou"
  reg_weights: 1.0

FCOSPostProcess:
  decode:
    name: FCOSBox
    num_classes: 80
  nms:
    name: MultiClassNMS
    nms_top_k: 1000
    keep_top_k: 100
    score_threshold: 0.025
    nms_threshold: 0.6