support xxxevaluator

open-mmlab · Sep 19, 2022 · 4dc2e44 · 4dc2e44
1 parent fe2cd4a
commit 4dc2e44
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/tools/test.py b/tools/test.py
@@ -108,8 +108,17 @@ def main():
         dump_metric = dict(type='DumpResults', out_file_path=args.dump)
         if isinstance(cfg.test_evaluator, (list, tuple)):
             cfg.test_evaluator = list(cfg.test_evaluator).append(dump_metric)
-        else:
-            cfg.test_evaluator = [cfg.test_evaluator, dump_metric]
+        elif isinstance(cfg.test_evaluator, dict):
+            if 'metric' in cfg.test_evaluator:
+                if isinstance(cfg.test_evaluator.metric, (list, tuple)):
+                    cfg.test_evaluator.metric = list(
+                        cfg.test_evaluator.metric).append(dump_metric)
+                else:
+                    cfg.test_evaluator.metric = [
+                        cfg.test_evaluator.metric, dump_metric
+                    ]
+            else:
+                cfg.test_evaluator = [cfg.test_evaluator, dump_metric]
 
     # build the runner from config
     if 'runner_type' not in cfg: