rpn_r50_fpn_1x_coco.py 1.1 KB

123456789101112131415161718192021222324252627282930313233343536
  1. _base_ = [
  2. '../_base_/models/rpn_r50_fpn.py', '../_base_/datasets/coco_detection.py',
  3. '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py'
  4. ]
  5. val_evaluator = dict(metric='proposal_fast')
  6. test_evaluator = val_evaluator
  7. # inference on val dataset and dump the proposals with evaluate metric
  8. # data_root = 'data/coco/'
  9. # test_evaluator = [
  10. # dict(
  11. # type='DumpProposals',
  12. # output_dir=data_root + 'proposals/',
  13. # proposals_file='rpn_r50_fpn_1x_val2017.pkl'),
  14. # dict(
  15. # type='CocoMetric',
  16. # ann_file=data_root + 'annotations/instances_val2017.json',
  17. # metric='proposal_fast',
  18. # backend_args={{_base_.backend_args}},
  19. # format_only=False)
  20. # ]
  21. # inference on training dataset and dump the proposals without evaluate metric
  22. # data_root = 'data/coco/'
  23. # test_dataloader = dict(
  24. # dataset=dict(
  25. # ann_file='annotations/instances_train2017.json',
  26. # data_prefix=dict(img='train2017/')))
  27. #
  28. # test_evaluator = [
  29. # dict(
  30. # type='DumpProposals',
  31. # output_dir=data_root + 'proposals/',
  32. # proposals_file='rpn_r50_fpn_1x_train2017.pkl'),
  33. # ]