未验证 提交 5ee9a605 编写于 作者: W wangguanzhong 提交者: GitHub

fix empty output in voc (#3255)

上级 bcdae9c6
...@@ -235,9 +235,10 @@ class VOCMetric(Metric): ...@@ -235,9 +235,10 @@ class VOCMetric(Metric):
self.detection_map.reset() self.detection_map.reset()
def update(self, inputs, outputs): def update(self, inputs, outputs):
bboxes = outputs['bbox'][:, 2:].numpy() bbox_np = outputs['bbox'].numpy()
scores = outputs['bbox'][:, 1].numpy() bboxes = bbox_np[:, 2:]
labels = outputs['bbox'][:, 0].numpy() scores = bbox_np[:, 1]
labels = bbox_np[:, 0]
bbox_lengths = outputs['bbox_num'].numpy() bbox_lengths = outputs['bbox_num'].numpy()
if bboxes.shape == (1, 1) or bboxes is None: if bboxes.shape == (1, 1) or bboxes is None:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册