56 lines
1.1 KiB
YAML
56 lines
1.1 KiB
YAML
architecture: S2ANet
|
|
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_cos_pretrained.pdparams
|
|
weights: output/s2anet_r50_fpn_1x_dota/model_final.pdparams
|
|
|
|
|
|
# Model Achitecture
|
|
S2ANet:
|
|
backbone: ResNet
|
|
neck: FPN
|
|
s2anet_head: S2ANetHead
|
|
s2anet_bbox_post_process: S2ANetBBoxPostProcess
|
|
|
|
ResNet:
|
|
depth: 50
|
|
norm_type: bn
|
|
return_idx: [1,2,3]
|
|
num_stages: 4
|
|
|
|
FPN:
|
|
in_channels: [256, 512, 1024]
|
|
out_channel: 256
|
|
spatial_scales: [0.25, 0.125, 0.0625]
|
|
has_extra_convs: True
|
|
extra_stage: 2
|
|
relu_before_extra_convs: False
|
|
|
|
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: 15
|
|
align_conv_type: 'Conv' # AlignConv Conv
|
|
align_conv_size: 3
|
|
use_sigmoid_cls: True
|
|
|
|
RBoxAssigner:
|
|
pos_iou_thr: 0.5
|
|
neg_iou_thr: 0.4
|
|
min_iou_thr: 0.0
|
|
ignore_iof_thr: -2
|
|
|
|
S2ANetBBoxPostProcess:
|
|
nms_pre: 2000
|
|
min_bbox_size: 0.0
|
|
nms:
|
|
name: MultiClassNMS
|
|
keep_top_k: -1
|
|
score_threshold: 0.05
|
|
nms_threshold: 0.1
|
|
normalized: False
|
|
#background_label: -1
|