diff --git a/python-package/insightface/app/face_analysis.py b/python-package/insightface/app/face_analysis.py index 6c3dfcbd2..a9112b14a 100644 --- a/python-package/insightface/app/face_analysis.py +++ b/python-package/insightface/app/face_analysis.py @@ -55,10 +55,10 @@ def prepare(self, ctx_id, det_thresh=0.5, det_size=(640, 640)): else: model.prepare(ctx_id) - def get(self, img, max_num=0): + def get(self, img, max_num=0, det_metric='default'): bboxes, kpss = self.det_model.detect(img, max_num=max_num, - metric='default') + metric=det_metric) if bboxes.shape[0] == 0: return [] ret = []