From ffa12d315041c0505b077c7ff057ca914bb8dc7e Mon Sep 17 00:00:00 2001 From: Jia Guo Date: Sat, 17 Dec 2022 10:13:05 +0800 Subject: [PATCH] fix align template while inputsize divided by 128 --- python-package/insightface/__init__.py | 2 +- python-package/insightface/utils/face_align.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/python-package/insightface/__init__.py b/python-package/insightface/__init__.py index ac680147..5530e848 100644 --- a/python-package/insightface/__init__.py +++ b/python-package/insightface/__init__.py @@ -11,7 +11,7 @@ except ImportError: "Unable to import dependency onnxruntime. " ) -__version__ = '0.7.1' +__version__ = '0.7.2' from . import model_zoo from . import utils diff --git a/python-package/insightface/utils/face_align.py b/python-package/insightface/utils/face_align.py index 57f90f92..226628b3 100644 --- a/python-package/insightface/utils/face_align.py +++ b/python-package/insightface/utils/face_align.py @@ -13,9 +13,12 @@ def estimate_norm(lmk, image_size=112,mode='arcface'): assert image_size%112==0 or image_size%128==0 if image_size%112==0: ratio = float(image_size)/112.0 + diff_x = 0 else: ratio = float(image_size)/128.0 + diff_x = 8.0*ratio dst = arcface_dst * ratio + dst[:,0] += diff_x tform = trans.SimilarityTransform() tform.estimate(lmk, dst) M = tform.params[0:2, :]