ade20k_panoptic.py 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. # dataset settings
  2. dataset_type = 'ADE20KPanopticDataset'
  3. data_root = 'data/ADEChallengeData2016/'
  4. backend_args = None
  5. test_pipeline = [
  6. dict(type='LoadImageFromFile', backend_args=backend_args),
  7. dict(type='Resize', scale=(2560, 640), keep_ratio=True),
  8. dict(type='LoadPanopticAnnotations', backend_args=backend_args),
  9. dict(
  10. type='PackDetInputs',
  11. meta_keys=('img_id', 'img_path', 'ori_shape', 'img_shape',
  12. 'scale_factor'))
  13. ]
  14. val_dataloader = dict(
  15. batch_size=1,
  16. num_workers=0,
  17. persistent_workers=False,
  18. drop_last=False,
  19. sampler=dict(type='DefaultSampler', shuffle=False),
  20. dataset=dict(
  21. type=dataset_type,
  22. data_root=data_root,
  23. ann_file='ade20k_panoptic_val.json',
  24. data_prefix=dict(img='images/validation/', seg='ade20k_panoptic_val/'),
  25. test_mode=True,
  26. pipeline=test_pipeline,
  27. backend_args=backend_args))
  28. test_dataloader = val_dataloader
  29. val_evaluator = dict(
  30. type='CocoPanopticMetric',
  31. ann_file=data_root + 'ade20k_panoptic_val.json',
  32. seg_prefix=data_root + 'ade20k_panoptic_val/',
  33. backend_args=backend_args)
  34. test_evaluator = val_evaluator