[Fix] Fix an error on indexing scalar metrics in analyze_result.py (o…

…pen-mmlab#518) * [Fix] Error on indexing scalar metrics * [Fix] Error on indexing scalar metrics Co-authored-by: Ma Zerun <mzr1996@163.com> Co-authored-by: Ma Zerun <mzr1996@163.com>
mzr1996 · Nov 5, 2021 · ee307d4 · ee307d4
1 parent 9bb1103
commit ee307d4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tools/analysis_tools/analyze_results.py b/tools/analysis_tools/analyze_results.py
@@ -97,6 +97,11 @@ def main():
     outputs['gt_label'] = gt_labels
     outputs['gt_class'] = gt_classes
 
+    need_keys = [
+        'filename', 'gt_label', 'gt_class', 'pred_score', 'pred_label',
+        'pred_class'
+    ]
+    outputs = {k: v for k, v in outputs.items() if k in need_keys}
     outputs_list = list()
     for i in range(len(gt_labels)):
         output = dict()