[Bugfix] analyze_results (#10876)

open-mmlab · Sep 4, 2023 · a9c4c01 · a9c4c01
1 parent 3bccdaf
commit a9c4c01
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/tools/analysis_tools/analyze_results.py b/tools/analysis_tools/analyze_results.py
@@ -124,8 +124,12 @@ def _save_image_gts_results(self,
 
             if task == 'det':
                 gt_instances = InstanceData()
-                gt_instances.bboxes = results[index]['gt_instances']['bboxes']
-                gt_instances.labels = results[index]['gt_instances']['labels']
+                gt_instances.bboxes = [
+                    d['bbox'] for d in data_info['gt_instances']
+                ]
+                gt_instances.labels = [
+                    d['bbox_label'] for d in data_info['gt_instances']
+                ]
 
                 pred_instances = InstanceData()
                 pred_instances.bboxes = results[index]['pred_instances'][
@@ -141,7 +145,9 @@ def _save_image_gts_results(self,
 
             elif task == 'seg':
                 gt_panoptic_seg = PixelData()
-                gt_panoptic_seg.sem_seg = results[index]['gt_seg_map']
+                gt_panoptic_seg.sem_seg = [
+                    d['gt_seg_map'] for d in data_info['gt_instances']
+                ]
 
                 pred_panoptic_seg = PixelData()
                 pred_panoptic_seg.sem_seg = results[index][