Add CityScapes evaluation as fallback option for datasets

lxdo517 · Sep 15, 2019 · 5a99c70 · 5a99c70
1 parent 8020d87
commit 5a99c70
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/maskrcnn_benchmark/data/datasets/evaluation/__init__.py b/maskrcnn_benchmark/data/datasets/evaluation/__init__.py
@@ -2,7 +2,7 @@
 
 from .coco import coco_evaluation
 from .voc import voc_evaluation
-
+from .cityscapes import abs_cityscapes_evaluation
 
 def evaluate(dataset, predictions, output_folder, **kwargs):
     """evaluate dataset using different methods based on dataset type.
@@ -22,6 +22,8 @@ def evaluate(dataset, predictions, output_folder, **kwargs):
         return coco_evaluation(**args)
     elif isinstance(dataset, datasets.PascalVOCDataset):
         return voc_evaluation(**args)
+    elif isinstance(dataset, datasets.AbstractDataset):
+        return abs_cityscapes_evaluation(**args)
     else:
         dataset_name = dataset.__class__.__name__
         raise NotImplementedError("Unsupported dataset type {}.".format(dataset_name))