From 98fb689530bc3669e5d2663764a2f52c0fe2a877 Mon Sep 17 00:00:00 2001
From: xudou3 <296566150@qq.com>
Date: Thu, 4 Nov 2021 15:21:28 +0800
Subject: [PATCH] debug picodet_postprocess when return none box (#4444)

---
 deploy/python/picodet_postprocess.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deploy/python/picodet_postprocess.py b/deploy/python/picodet_postprocess.py
index 1fed8c641..f5911a697 100644
--- 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
-- 
GitLab