htc_x101-64x4d-dconv-c3-c5_fpn_ms-400-1400-16xb1-20e_coco.py 616 B

1234567891011121314151617181920
  1. _base_ = './htc_x101-64x4d_fpn_16xb1-20e_coco.py'
  2. model = dict(
  3. backbone=dict(
  4. dcn=dict(type='DCN', deform_groups=1, fallback_on_stride=False),
  5. stage_with_dcn=(False, True, True, True)))
  6. # dataset settings
  7. train_pipeline = [
  8. dict(type='LoadImageFromFile'),
  9. dict(
  10. type='LoadAnnotations', with_bbox=True, with_mask=True, with_seg=True),
  11. dict(
  12. type='RandomResize',
  13. scale=[(1600, 400), (1600, 1400)],
  14. keep_ratio=True),
  15. dict(type='RandomFlip', prob=0.5),
  16. dict(type='PackDetInputs')
  17. ]
  18. train_dataloader = dict(dataset=dict(pipeline=train_pipeline))