Browse Source

fix empty output in voc

will-jl944 4 năm trước cách đây
mục cha
commit
76f4f777bf
1 tập tin đã thay đổi với 4 bổ sung3 xóa
  1. 4 3
      dygraph/paddlex/cv/models/utils/det_metrics/metrics.py

+ 4 - 3
dygraph/paddlex/cv/models/utils/det_metrics/metrics.py

@@ -85,9 +85,10 @@ class VOCMetric(Metric):
         self.detection_map.reset()
 
     def update(self, inputs, outputs):
-        bboxes = outputs['bbox'][:, 2:].numpy()
-        scores = outputs['bbox'][:, 1].numpy()
-        labels = outputs['bbox'][:, 0].numpy()
+        bbox_np = outputs['bbox'].numpy()
+        bboxes = bbox_np[:, 2:]
+        scores = bbox_np[:, 1]
+        labels = bbox_np[:, 0]
         bbox_lengths = outputs['bbox_num'].numpy()
 
         if bboxes.shape == (1, 1) or bboxes is None: