coco.py 988 B

123456789101112131415161718192021222324252627282930313233
  1. _base_ = [
  2. '../_base_/models/faster-rcnn_r50_fpn.py',
  3. '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py',
  4. '../_base_/datasets/dsdl.py'
  5. ]
  6. # dsdl dataset settings
  7. # please visit our platform [OpenDataLab](https://opendatalab.com/)
  8. # to downloaded dsdl dataset.
  9. data_root = 'data/COCO2017'
  10. img_prefix = 'original'
  11. train_ann = 'dsdl/set-train/train.yaml'
  12. val_ann = 'dsdl/set-val/val.yaml'
  13. specific_key_path = dict(ignore_flag='./annotations/*/iscrowd')
  14. train_dataloader = dict(
  15. dataset=dict(
  16. specific_key_path=specific_key_path,
  17. data_root=data_root,
  18. ann_file=train_ann,
  19. data_prefix=dict(img_path=img_prefix),
  20. filter_cfg=dict(filter_empty_gt=True, min_size=32, bbox_min_size=32),
  21. ))
  22. val_dataloader = dict(
  23. dataset=dict(
  24. specific_key_path=specific_key_path,
  25. data_root=data_root,
  26. ann_file=val_ann,
  27. data_prefix=dict(img_path=img_prefix),
  28. ))
  29. test_dataloader = val_dataloader