pip change to 0.3.4

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

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):

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):

python-package/insightface/utils/constant.py

@@ -1,3 +1,3 @@
 
-DEFAULT_MP_NAME = 'antelope'
+DEFAULT_MP_NAME = 'antelopev2'