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

Update model_zoo.py

Browse Source
This commit is contained in:
신관준
2022-01-11 03:06:20 +09:00
committed by GitHub
parent 983af548d1
commit 11590781c4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -49,7 +49,7 @@ class ModelRouter:
self.onnx_file = onnx_file
def get_model(self, **kwargs):
session = onnxruntime.InferenceSession(self.onnx_file, **kwargs)
session = PickableInferenceSession(self.onnx_file, **kwargs)
print(f'Applied providers: {session._providers}, with options: {session._provider_options}')
input_cfg = session.get_inputs()[0]
input_shape = input_cfg.shape