Browse Source

Merge pull request #856 from will-jl944/develop_jf

fix empty output in voc
FlyingQianMM 4 years ago
parent
commit
13f91b128b
1 changed files with 4 additions and 3 deletions
  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: