From 912904c4e8e179aea4bcf692ba5be17dd167ab3d Mon Sep 17 00:00:00 2001 From: TonyCongqianWang <35334345+TonyCongqianWang@users.noreply.github.com> Date: Thu, 18 Jul 2024 06:21:16 +0200 Subject: [PATCH] fixing bug crashing eval on widerface (#267) Co-authored-by: Tony Wang --- models/face_detection_yunet/yunet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/face_detection_yunet/yunet.py b/models/face_detection_yunet/yunet.py index 21558255..710d24b8 100644 --- a/models/face_detection_yunet/yunet.py +++ b/models/face_detection_yunet/yunet.py @@ -52,4 +52,4 @@ def setInputSize(self, input_size): def infer(self, image): # Forward faces = self._model.detect(image) - return np.array([]) if faces[1] is None else faces[1] + return np.empty(shape=(0, 5)) if faces[1] is None else faces[1]