未验证 提交 e3029fd8 编写于 作者: C cnn 提交者: GitHub

[dev] clean code of s2anet (#3733) (#3778)

* clean code of s2anet

* fix for paddle==2.1.0

* remove debug info

* update comment
上级 621d2bdd
architecture: S2ANet
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_cos_pretrained.pdparams
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams
weights: output/s2anet_r50_fpn_1x_dota/model_final.pdparams
......@@ -12,6 +12,7 @@ S2ANet:
ResNet:
depth: 50
variant: d
norm_type: bn
return_idx: [1,2,3]
num_stages: 4
......@@ -33,11 +34,9 @@ S2ANetHead:
feat_in: 256
feat_out: 256
num_classes: 15
align_conv_type: 'Conv' # AlignConv Conv
align_conv_type: 'AlignConv' # 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 ]
RBoxAssigner:
pos_iou_thr: 0.5
......@@ -54,3 +53,4 @@ S2ANetBBoxPostProcess:
score_threshold: 0.05
nms_threshold: 0.1
normalized: False
#background_label: -1
此差异已折叠。
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册