_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))) | |