diff --git a/python-package/insightface/__init__.py b/python-package/insightface/__init__.py index 8af9c90..a173a36 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.3.3' +__version__ = '0.3.4' from . import model_zoo from . import utils diff --git a/python-package/insightface/app/face_analysis.py b/python-package/insightface/app/face_analysis.py index 1bf275e..21cd9dc 100644 --- a/python-package/insightface/app/face_analysis.py +++ b/python-package/insightface/app/face_analysis.py @@ -34,7 +34,9 @@ class FaceAnalysis: #print('ignore:', onnx_file) continue model = model_zoo.get_model(onnx_file) - if allowed_modules is not None and model.taskname not in allowed_modules: + if model is None: + print('model not recognized:', onnx_file) + elif allowed_modules is not None and model.taskname not in allowed_modules: print('model ignore:', onnx_file, model.taskname) del model elif model.taskname not in self.models and (allowed_modules is None or model.taskname in allowed_modules): diff --git a/python-package/insightface/model_zoo/model_zoo.py b/python-package/insightface/model_zoo/model_zoo.py index 809e21c..7aeec1f 100644 --- a/python-package/insightface/model_zoo/model_zoo.py +++ b/python-package/insightface/model_zoo/model_zoo.py @@ -36,7 +36,8 @@ class ModelRouter: elif input_shape[2]==96 and input_shape[3]==96: return Attribute(model_file=self.onnx_file, session=session) else: - raise RuntimeError('error on model routing') + #raise RuntimeError('error on model routing') + return None def find_onnx_file(dir_path): if not os.path.exists(dir_path): diff --git a/python-package/insightface/utils/constant.py b/python-package/insightface/utils/constant.py index 65ec088..6f52502 100644 --- a/python-package/insightface/utils/constant.py +++ b/python-package/insightface/utils/constant.py @@ -1,3 +1,3 @@ -DEFAULT_MP_NAME = 'antelope' +DEFAULT_MP_NAME = 'antelopev2'