| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091 |
- architecture: MaskRCNN
- pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_cos_pretrained.pdparams
- MaskRCNN:
- backbone: ResNet
- neck: FPN
- rpn_head: RPNHead
- bbox_head: BBoxHead
- mask_head: MaskHead
- # post process
- bbox_post_process: BBoxPostProcess
- mask_post_process: MaskPostProcess
- ResNet:
- # index 0 stands for res2
- depth: 50
- norm_type: bn
- freeze_at: 0
- return_idx: [0,1,2,3]
- num_stages: 4
- FPN:
- out_channel: 256
- RPNHead:
- anchor_generator:
- aspect_ratios: [0.5, 1.0, 2.0]
- anchor_sizes: [[32], [64], [128], [256], [512]]
- strides: [4, 8, 16, 32, 64]
- rpn_target_assign:
- batch_size_per_im: 256
- fg_fraction: 0.5
- negative_overlap: 0.3
- positive_overlap: 0.7
- use_random: True
- train_proposal:
- min_size: 0.0
- nms_thresh: 0.7
- pre_nms_top_n: 2000
- post_nms_top_n: 1000
- topk_after_collect: True
- test_proposal:
- min_size: 0.0
- nms_thresh: 0.7
- pre_nms_top_n: 1000
- post_nms_top_n: 1000
- BBoxHead:
- head: TwoFCHead
- roi_extractor:
- resolution: 7
- sampling_ratio: 0
- aligned: True
- bbox_assigner: BBoxAssigner
- BBoxAssigner:
- batch_size_per_im: 512
- bg_thresh: 0.5
- fg_thresh: 0.5
- fg_fraction: 0.25
- use_random: True
- TwoFCHead:
- out_channel: 1024
- BBoxPostProcess:
- decode: RCNNBox
- nms:
- name: MultiClassNMS
- keep_top_k: 100
- score_threshold: 0.05
- nms_threshold: 0.5
- MaskHead:
- head: MaskFeat
- roi_extractor:
- resolution: 14
- sampling_ratio: 0
- aligned: True
- mask_assigner: MaskAssigner
- share_bbox_feat: False
- MaskFeat:
- num_convs: 4
- out_channel: 256
- MaskAssigner:
- mask_resolution: 28
- MaskPostProcess:
- binary_thresh: 0.5
|
