mask-rcnn_r50_fpn_poly-1x_coco.py 581 B

123456789101112131415161718
  1. _base_ = [
  2. '../_base_/models/mask-rcnn_r50_fpn.py',
  3. '../_base_/datasets/coco_instance.py',
  4. '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py'
  5. ]
  6. train_pipeline = [
  7. dict(type='LoadImageFromFile', backend_args={{_base_.backend_args}}),
  8. dict(
  9. type='LoadAnnotations',
  10. with_bbox=True,
  11. with_mask=True,
  12. poly2mask=False),
  13. dict(type='Resize', scale=(1333, 800), keep_ratio=True),
  14. dict(type='RandomFlip', prob=0.5),
  15. dict(type='PackDetInputs'),
  16. ]
  17. train_dataloader = dict(dataset=dict(pipeline=train_pipeline))