Update face_analysis.py

rename param name.
This commit is contained in:
ease_zh
2024-11-01 11:28:46 +08:00
committed by GitHub
parent 00595a7c58
commit a333044b41

View File

@@ -55,10 +55,10 @@ class FaceAnalysis:
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 = []