|
_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( |
|
roi_head=dict( |
|
bbox_head=dict(num_classes=10), mask_head=dict(num_classes=10))) |
|
|
|
backend_args = None |
|
|
|
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_root = 'data/nuimages/' |
|
|
|
|
|
|
|
|
|
|
|
|
