From cdc3d4ed5de14712378f3d5a14249661e54a03ec Mon Sep 17 00:00:00 2001
From: Jia Guo <guojia@gmail.com>
Date: Mon, 23 Aug 2021 19:33:51 +0800
Subject: [PATCH] package 0.4.1

---
 python-package/insightface/__init__.py         | 2 +-
 python-package/insightface/utils/face_align.py | 6 ++++--
 python-package/setup.py                        | 4 ++++
 3 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/python-package/insightface/__init__.py b/python-package/insightface/__init__.py
index 76386f6..0577356 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.4'
+__version__ = '0.4.1'
 
 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 dcbb099..c4e9a7c 100644
--- a/python-package/insightface/utils/face_align.py
+++ b/python-package/insightface/utils/face_align.py
@@ -47,8 +47,10 @@ def estimate_norm(lmk, image_size=112, mode='arcface'):
     min_index = []
     min_error = float('inf')
     if mode == 'arcface':
-        assert image_size == 112
-        src = arcface_src
+        if image_size == 112:
+            src = arcface_src
+        else:
+            src = float(image_size) / 112 * arcface_src
     else:
         src = src_map[image_size]
     for i in np.arange(src.shape[0]):
diff --git a/python-package/setup.py b/python-package/setup.py
index 4229664..316f6c3 100644
--- a/python-package/setup.py
+++ b/python-package/setup.py
@@ -24,11 +24,13 @@ def find_version(*file_paths):
         return version_match.group(1)
     raise RuntimeError("Unable to find version string.")
 
+pypandoc_enabled = True
 try:
     import pypandoc
     long_description = pypandoc.convert('README.md', 'rst')
 except (IOError, ImportError, ModuleNotFoundError):
     long_description = open('README.md').read()
+    pypandoc_enabled = False
 
 #import pypandoc
 #long_description = pypandoc.convert('README.md', 'rst')
@@ -88,3 +90,5 @@ setup(
     include_dirs=numpy.get_include(),
 )
 
+print('pypandoc enabled:', pypandoc_enabled)
+