Merge pull request deepinsight#2679 from ease-zh/patch-1

Update face_analysis.py
truststamp · Dec 5, 2024 · 4aa1a40 · 4aa1a40
2 parents 2e29b41 + a333044
commit 4aa1a40
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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 = []