ZF/insightface
1
0
Fork 0
mirror of https://github.com/deepinsight/insightface.git synced 2025-12-30 08:02:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

bug fix

Browse Source
This commit is contained in:
Jia Guo
2021-07-07 23:17:43 +08:00
parent cc488b94a6
commit 2400a75dd0
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python-package/insightface/app/face_analysis.py
View File

@@ -12,6 +12,7 @@ import glob
import os
import os.path as osp
from numpy.linalg import norm
import onnxruntime
from ..model_zoo import model_zoo
from ..utils import face_align
from ..utils import ensure_available
@@ -22,11 +23,12 @@ __all__ = ['FaceAnalysis']
class FaceAnalysis:
def __init__(self, name=DEFAULT_MP_NAME, root='~/.insightface/models', allowed_modules=None):
onnxruntime.set_default_logger_severity(3)
self.models = {}
#root = os.path.expanduser(root)
#self.model_dir = osp.join(root, name)
#self.model_dir = get_model_dir(name, root)
self.model_dir = ensure_available('models', name)
self.model_dir = ensure_available('models', name, root=root)
onnx_files = glob.glob(osp.join(self.model_dir, '*.onnx'))
onnx_files = sorted(onnx_files)
for onnx_file in onnx_files:

2
python-package/insightface/model_zoo/arcface_onnx.py
View File

@@ -68,7 +68,7 @@ class ArcFaceONNX:
return face.embedding
def compute_sim(self, feat1, feat2):
from np.linalg import norm
from numpy.linalg import norm
feat1 = feat1.ravel()
feat2 = feat2.ravel()
sim = np.dot(feat1, feat2) / (norm(feat1) * norm(feat2))