Skip to content
Snippets Groups Projects
Commit 7c329fa7 authored by sshuair's avatar sshuair Committed by GitHub
Browse files

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
parent 71be1c27
No related branches found
No related tags found
No related merge requests found
......@@ -374,6 +374,11 @@ class CustomDataset(Dataset):
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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment