diff --git a/insightface_func/face_detect_crop_multi.py b/insightface_func/face_detect_crop_multi.py index 1c5035a..3b75b9e 100644 --- a/insightface_func/face_detect_crop_multi.py +++ b/insightface_func/face_detect_crop_multi.py @@ -49,7 +49,7 @@ class Face_detect_crop: def prepare(self, ctx_id, det_thresh=0.5, det_size=(640, 640), mode ='None'): - self.det_thresh = det_thresh + self.det_model.det_thresh = det_thresh self.mode = mode assert det_size is not None print('set det-size:', det_size) diff --git a/insightface_func/face_detect_crop_single.py b/insightface_func/face_detect_crop_single.py index 9ab7a9b..9061d47 100644 --- a/insightface_func/face_detect_crop_single.py +++ b/insightface_func/face_detect_crop_single.py @@ -49,7 +49,7 @@ class Face_detect_crop: def prepare(self, ctx_id, det_thresh=0.5, det_size=(640, 640), mode ='None'): - self.det_thresh = det_thresh + self.det_model.det_thresh = det_thresh self.mode = mode assert det_size is not None print('set det-size:', det_size)