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