Fix TypeError: RetinaFace.detect() got an unexpected keyword argument 'threshold'

woctezuma · woctezuma · commit 061329137d6d · 2023-09-21T00:15:52.000+02:00
diff --git a/insightface_func/face_detect_crop_multi.py b/insightface_func/face_detect_crop_multi.py
@@ -62,7 +62,6 @@ def prepare(self, ctx_id, det_thresh=0.5, det_size=(640, 640), mode ='None'):
 
     def get(self, img, crop_size, max_num=0):
         bboxes, kpss = self.det_model.detect(img,
-                                             threshold=self.det_thresh,
                                              max_num=max_num,
                                              metric='default')
         if bboxes.shape[0] == 0:
diff --git a/insightface_func/face_detect_crop_single.py b/insightface_func/face_detect_crop_single.py
@@ -62,7 +62,6 @@ def prepare(self, ctx_id, det_thresh=0.5, det_size=(640, 640), mode ='None'):
 
     def get(self, img, crop_size, max_num=0):
         bboxes, kpss = self.det_model.detect(img,
-                                             threshold=self.det_thresh,
                                              max_num=max_num,
                                              metric='default')
         if bboxes.shape[0] == 0:
