Johannes commited on
Commit
d2cbc11
·
1 Parent(s): fd4d7a2

fix kornia batch problem

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -123,7 +123,7 @@ def kornia_detect(img, confidence_threshold, nms_threshold):
123
 
124
  with torch.no_grad():
125
  dets = face_detection(img)
126
- dets = [FaceDetectorResult(o) for o in dets]
127
 
128
  # show image
129
 
 
123
 
124
  with torch.no_grad():
125
  dets = face_detection(img)
126
+ dets = [FaceDetectorResult(o) for o in dets[0]]
127
 
128
  # show image
129