Update face_analysis.py

rename param name.
truststamp · Nov 1, 2024 · a333044 · a333044
1 parent 00595a7
commit a333044
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, metric='default'):
+    def get(self, img, max_num=0, det_metric='default'):
         bboxes, kpss = self.det_model.detect(img,
                                              max_num=max_num,
-                                             metric=metric)
+                                             metric=det_metric)
         if bboxes.shape[0] == 0:
             return []
         ret = []