update Google FaceMesh model to the latest version

iperov · Nov 12, 2021 · 894ec56 · 894ec56
1 parent 6978619
commit 894ec56
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/modelhub/onnx/FaceMesh/FaceMesh.onnx b/modelhub/onnx/FaceMesh/FaceMesh.onnx
diff --git a/modelhub/onnx/FaceMesh/FaceMesh.py b/modelhub/onnx/FaceMesh/FaceMesh.py
@@ -30,6 +30,9 @@ def __init__(self, device_info : ORTDeviceInfo):
             raise Exception(f'device_info {device_info} is not in available devices for FaceMesh')
 
         path = Path(__file__).parent / 'FaceMesh.onnx'
+        if not path.exists():
+            raise FileNotFoundError(f'{path} not found')
+
         self._sess = sess = InferenceSession_with_device(str(path), device_info)
         self._input_name = sess.get_inputs()[0].name
         self._input_width = 192