| _base_ = [ | |
| '../_base_/models/mask-rcnn_r50_fpn.py', | |
| '../_base_/datasets/nuim-instance.py', | |
| '../_base_/schedules/mmdet-schedule-1x.py', '../_base_/default_runtime.py' | |
| ] | |
| model = dict( | |
| pretrained='open-mmlab://detectron2/resnet50_caffe', | |
| backbone=dict(norm_cfg=dict(requires_grad=False), style='caffe'), | |
| roi_head=dict( | |
| bbox_head=dict(num_classes=10), mask_head=dict(num_classes=10))) | |
| backend_args = None | |
| train_pipeline = [ | |
| dict(type='LoadImageFromFile', backend_args=backend_args), | |
| dict(type='LoadAnnotations', with_bbox=True, with_mask=True), | |
| dict( | |
| type='Resize', | |
| img_scale=[(1280, 720), (1920, 1080)], | |
| multiscale_mode='range', | |
| keep_ratio=True), | |
| dict(type='RandomFlip', flip_ratio=0.5), | |
| dict(type='PackDetInputs'), | |
| ] | |
| test_pipeline = [ | |
| dict(type='LoadImageFromFile', backend_args=backend_args), | |
| dict( | |
| type='MultiScaleFlipAug', | |
| img_scale=(1600, 900), | |
| flip=False, | |
| transforms=[ | |
| dict(type='Resize', keep_ratio=True), | |
| dict(type='RandomFlip'), | |
| ]), | |
| dict( | |
| type='PackDetInputs', | |
| meta_keys=('img_id', 'img_path', 'ori_shape', 'img_shape', | |
| 'scale_factor')), | |
| ] | |
| data = dict( | |
| train=dict(pipeline=train_pipeline), | |
| val=dict(pipeline=test_pipeline), | |
| test=dict(pipeline=test_pipeline)) | |