Include each class metrics in logs (#445)

* Include each class metrics in logs * format * fix the unitest * fix the custom int class_name * fix the custom int class_name
open-mmlab · Mar 29, 2021 · f7a5d53 · f7a5d53
1 parent 1fc3e37
commit f7a5d53
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/mmseg/datasets/custom.py b/mmseg/datasets/custom.py
@@ -374,6 +374,11 @@ def evaluate(self,
         for i in range(1, len(summary_table_data[0])):
             eval_results[summary_table_data[0]
                          [i]] = summary_table_data[1][i] / 100.0
+        for idx, sub_metric in enumerate(class_table_data[0][1:], 1):
+            for item in class_table_data[1:]:
+                eval_results[str(sub_metric) + '.' +
+                             str(item[0])] = item[idx] / 100.0
+
         if mmcv.is_list_of(results, str):
             for file_name in results:
                 os.remove(file_name)