diff --git a/configs/solov2/solov2_r101_vd_fpn_3x.yml b/configs/solov2/solov2_r101_vd_fpn_3x.yml new file mode 100644 index 0000000000000000000000000000000000000000..58f70ada17805b859a49be3f9e53c86540c9bab4 --- /dev/null +++ b/configs/solov2/solov2_r101_vd_fpn_3x.yml @@ -0,0 +1,95 @@ +architecture: SOLOv2 +use_gpu: true +max_iters: 270000 +snapshot_iter: 30000 +log_smooth_window: 20 +save_dir: output +pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/ResNet101_vd_pretrained.tar +metric: COCO +weights: output/solov2_r101_vd_fpn_dcn_multiscale_3x/model_final +num_classes: 81 +use_ema: true +ema_decay: 0.9998 + +SOLOv2: + backbone: ResNet + fpn: FPN + bbox_head: SOLOv2Head + mask_head: SOLOv2MaskHead + +ResNet: + depth: 101 + feature_maps: [2, 3, 4, 5] + freeze_at: 2 + norm_type: bn + dcn_v2_stages: [3, 4, 5] + variant: d + +FPN: + max_level: 6 + min_level: 2 + num_chan: 256 + spatial_scale: [0.03125, 0.0625, 0.125, 0.25] + reverse_out: True + +SOLOv2Head: + seg_feat_channels: 512 + stacked_convs: 4 + num_grids: [40, 36, 24, 16, 12] + kernel_out_channels: 256 + dcn_v2_stages: [0, 1, 2, 3] + +SOLOv2MaskHead: + out_channels: 128 + start_level: 0 + end_level: 3 + num_classes: 256 + use_dcn_in_tower: True + +LearningRate: + base_lr: 0.01 + schedulers: + - !PiecewiseDecay + gamma: 0.1 + milestones: [180000, 240000] + - !LinearWarmup + start_factor: 0. + steps: 1000 + +OptimizerBuilder: + optimizer: + momentum: 0.9 + type: Momentum + regularizer: + factor: 0.0001 + type: L2 + +_READER_: 'solov2_reader.yml' +TrainReader: + sample_transforms: + - !DecodeImage + to_rgb: true + - !Poly2Mask {} + - !ResizeImage + target_size: [640, 672, 704, 736, 768, 800] + max_size: 1333 + interp: 1 + use_cv2: true + resize_box: true + - !RandomFlipImage + prob: 0.5 + - !NormalizeImage + is_channel_first: false + is_scale: true + mean: [0.485,0.456,0.406] + std: [0.229, 0.224,0.225] + - !Permute + to_bgr: false + channel_first: true + batch_transforms: + - !PadBatch + pad_to_stride: 32 + - !Gt2Solov2Target + num_grids: [40, 36, 24, 16, 12] + scale_ranges: [[1, 96], [48, 192], [96, 384], [192, 768], [384, 2048]] + coord_sigma: 0.2 diff --git a/configs/solov2/solov2_r50_fpn_1x.yml b/configs/solov2/solov2_r50_fpn_1x.yml index f7f26475a3b43803fded5c6d8304f74d35e26a30..af38b4bd5f79b85ca1c365df352beab0c3363016 100644 --- a/configs/solov2/solov2_r50_fpn_1x.yml +++ b/configs/solov2/solov2_r50_fpn_1x.yml @@ -14,7 +14,6 @@ SOLOv2: fpn: FPN bbox_head: SOLOv2Head mask_head: SOLOv2MaskHead - batch_size: 2 ResNet: depth: 50 diff --git a/configs/solov2/solov2_r50_fpn_3x.yml b/configs/solov2/solov2_r50_fpn_3x.yml index ee1b38eba070377c732ced4440fce319bdabc274..0cdf01f48c132bdbd0ba650e146e7b55a9b0d9e8 100644 --- a/configs/solov2/solov2_r50_fpn_3x.yml +++ b/configs/solov2/solov2_r50_fpn_3x.yml @@ -14,7 +14,6 @@ SOLOv2: fpn: FPN bbox_head: SOLOv2Head mask_head: SOLOv2MaskHead - batch_size: 2 ResNet: depth: 50