s2anet_conv_1x_dota.yml 637 字节
Newer Older
C
cnn 已提交
1
_BASE_: [
C
cnn 已提交
2
  '../datasets/dota.yml',
C
cnn 已提交
3 4 5 6 7 8
  '../runtime.yml',
  '_base_/s2anet_optimizer_1x.yml',
  '_base_/s2anet.yml',
  '_base_/s2anet_reader.yml',
]
weights: output/s2anet_1x_dota/model_final
C
cnn 已提交
9

C
cnn 已提交
10 11 12 13 14 15 16
ResNet:
  depth: 50
  variant: b
  norm_type: bn
  return_idx: [1,2,3]
  num_stages: 4

C
cnn 已提交
17
S2ANetHead:
C
cnn 已提交
18 19 20
  anchor_strides: [8, 16, 32, 64, 128]
  anchor_scales: [4]
  anchor_ratios: [1.0]
C
cnn 已提交
21 22 23 24 25 26 27 28
  anchor_assign: RBoxAssigner
  stacked_convs: 2
  feat_in: 256
  feat_out: 256
  num_classes: 15
  align_conv_type: 'Conv'  # AlignConv Conv
  align_conv_size: 3
  use_sigmoid_cls: True
29 30
  reg_loss_weight: [1.0, 1.0, 1.0, 1.0, 1.1]
  cls_loss_weight: [1.1, 1.05]