File size: 357 Bytes
3bbb319
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
_base_ = 'mask_rcnn_r50_fpg_crop640_50e_coco.py'

model = dict(
    neck=dict(out_channels=128, inter_channels=128),
    rpn_head=dict(in_channels=128),
    roi_head=dict(
        bbox_roi_extractor=dict(out_channels=128),
        bbox_head=dict(in_channels=128),
        mask_roi_extractor=dict(out_channels=128),
        mask_head=dict(in_channels=128)))