fast-rcnn_r50_fpn_1x_coco.py 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. _base_ = [
  2. '../_base_/models/fast-rcnn_r50_fpn.py',
  3. '../_base_/datasets/coco_detection.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(type='LoadProposals', num_max_proposals=2000),
  9. dict(type='LoadAnnotations', with_bbox=True),
  10. dict(
  11. type='ProposalBroadcaster',
  12. transforms=[
  13. dict(type='Resize', scale=(1333, 800), keep_ratio=True),
  14. dict(type='RandomFlip', prob=0.5),
  15. ]),
  16. dict(type='PackDetInputs')
  17. ]
  18. test_pipeline = [
  19. dict(type='LoadImageFromFile', backend_args={{_base_.backend_args}}),
  20. dict(type='LoadProposals', num_max_proposals=None),
  21. dict(
  22. type='ProposalBroadcaster',
  23. transforms=[
  24. dict(type='Resize', scale=(1333, 800), keep_ratio=True),
  25. ]),
  26. dict(
  27. type='PackDetInputs',
  28. meta_keys=('img_id', 'img_path', 'ori_shape', 'img_shape',
  29. 'scale_factor'))
  30. ]
  31. train_dataloader = dict(
  32. dataset=dict(
  33. proposal_file='proposals/rpn_r50_fpn_1x_train2017.pkl',
  34. pipeline=train_pipeline))
  35. val_dataloader = dict(
  36. dataset=dict(
  37. proposal_file='proposals/rpn_r50_fpn_1x_val2017.pkl',
  38. pipeline=test_pipeline))
  39. test_dataloader = val_dataloader