TonyCongqianWang Tony Wang commited on
Commit
489de70
·
1 Parent(s): f0ac01e

fixing bug crashing eval on widerface (#267)

Browse files

Co-authored-by: Tony Wang <[email protected]>

Files changed (1) hide show
  1. yunet.py +1 -1
yunet.py CHANGED
@@ -52,4 +52,4 @@ class YuNet:
52
  def infer(self, image):
53
  # Forward
54
  faces = self._model.detect(image)
55
- return np.array([]) if faces[1] is None else faces[1]
 
52
  def infer(self, image):
53
  # Forward
54
  faces = self._model.detect(image)
55
+ return np.empty(shape=(0, 5)) if faces[1] is None else faces[1]