s2anet_conv_2x_spine.yml 869 字节
Newer Older
1 2 3
_BASE_: [
  '../../../configs/datasets/spine_coco.yml',
  '../../../configs/runtime.yml',
4 5 6
  '../../../configs/rotate/s2anet/_base_/s2anet_optimizer_2x.yml',
  '../../../configs/rotate/s2anet/_base_/s2anet.yml',
  '../../../configs/rotate/s2anet/_base_/s2anet_reader.yml',
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
]
weights: output/s2anet_conv_1x_dota/model_final
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_cos_pretrained.pdparams

ResNet:
  depth: 50
  variant: b
  norm_type: bn
  return_idx: [1,2,3]
  num_stages: 4

S2ANetHead:
  anchor_strides: [8, 16, 32, 64, 128]
  anchor_scales: [4]
  anchor_ratios: [1.0]
  anchor_assign: RBoxAssigner
  stacked_convs: 2
  feat_in: 256
  feat_out: 256
  num_classes: 9
  align_conv_type: 'Conv'  # AlignConv Conv
  align_conv_size: 3
  use_sigmoid_cls: True
  reg_loss_weight: [1.0, 1.0, 1.0, 1.0, 1.1]
  cls_loss_weight: [1.1, 1.05]