Skip to content

Commit 912904c

Browse files
TonyCongqianWangTony Wang
and
Tony Wang
authored
fixing bug crashing eval on widerface (#267)
Co-authored-by: Tony Wang <Tony.Wang@mabri.vision>
1 parent f88e9b2 commit 912904c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/face_detection_yunet/yunet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,4 +52,4 @@ def setInputSize(self, input_size):
5252
def infer(self, image):
5353
# Forward
5454
faces = self._model.detect(image)
55-
return np.array([]) if faces[1] is None else faces[1]
55+
return np.empty(shape=(0, 5)) if faces[1] is None else faces[1]

0 commit comments

Comments
 (0)