未验证 提交 98fb6895 编写于 作者: X xudou3 提交者: GitHub

debug picodet_postprocess when return none box (#4444)

上级 3ef4f9d4
...@@ -194,7 +194,7 @@ class PicoDetPostProcess(object): ...@@ -194,7 +194,7 @@ class PicoDetPostProcess(object):
picked_box_probs.append(box_probs) picked_box_probs.append(box_probs)
picked_labels.extend([class_index] * box_probs.shape[0]) picked_labels.extend([class_index] * box_probs.shape[0])
if not picked_box_probs: if not picked_box_probs:
return np.array([]), np.array([]), np.array([]) return np.array([]), np.array([])
picked_box_probs = np.concatenate(picked_box_probs) picked_box_probs = np.concatenate(picked_box_probs)
# resize output boxes # resize output boxes
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册