debug picodet_postprocess when return none box (#4444)

98fb6895 · xudou3 · GitHub · 3ef4f9d4 · 98fb6895
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

deploy/python/picodet_postprocess.py deploy/python/picodet_postprocess.py +1 -1

未找到文件。
--- a/deploy/python/picodet_postprocess.py
+++ b/deploy/python/picodet_postprocess.py
@@ -194,7 +194,7 @@ class PicoDetPostProcess(object):
                picked_box_probs.append(box_probs)
                picked_labels.extend([class_index] * box_probs.shape[0])
            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)

            # resize output boxes