_base_ = [ | |
'../_base_/models/mask-rcnn_r50_fpn.py', | |
'../_base_/datasets/coco_instance.py', | |
'../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py' | |
] | |
train_pipeline = [ | |
dict(type='LoadImageFromFile', backend_args={{_base_.backend_args}}), | |
dict( | |
type='LoadAnnotations', | |
with_bbox=True, | |
with_mask=True, | |
poly2mask=False), | |
dict(type='Resize', scale=(1333, 800), keep_ratio=True), | |
dict(type='RandomFlip', prob=0.5), | |
dict(type='PackDetInputs'), | |
] | |
train_dataloader = dict(dataset=dict(pipeline=train_pipeline)) | |