From ac45dd43e49b098062a756c60e630ede8bd5e9a5 Mon Sep 17 00:00:00 2001 From: Wok Date: Thu, 21 Sep 2023 11:12:48 +0200 Subject: [PATCH] Fix bug: set det_thresh inside self.det_model --- insightface_func/face_detect_crop_multi.py | 2 +- insightface_func/face_detect_crop_single.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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)