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)