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]