| 1234567891011121314151617181920212223242526272829 |
- _BASE_: [
- '../datasets/spine_coco.yml',
- '../runtime.yml',
- '_base_/s2anet_optimizer_1x.yml',
- '_base_/s2anet.yml',
- '_base_/s2anet_reader.yml',
- ]
- weights: output/s2anet_1x_spine/model_final
- # for 8 card
- LearningRate:
- base_lr: 0.01
- 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: 'AlignConv' # AlignConv Conv
- align_conv_size: 3
- use_sigmoid_cls: True
- reg_loss_weight: [1.0, 1.0, 1.0, 1.0, 1.05]
- cls_loss_weight: [1.05, 1.0]
- reg_loss_type: 'l1'
|
