From 301d2f7facfcd89c6ac2f36d06fb37286ffd42b0 Mon Sep 17 00:00:00 2001 From: Avi Asulin Date: Tue, 31 Oct 2023 14:40:57 +0200 Subject: [PATCH 1/2] [mask-detection] add onnx missing requirements --- mask-detection/3-automatic-pipeline.ipynb | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/mask-detection/3-automatic-pipeline.ipynb b/mask-detection/3-automatic-pipeline.ipynb index eb6d5d56..6bcbff03 100644 --- a/mask-detection/3-automatic-pipeline.ipynb +++ b/mask-detection/3-automatic-pipeline.ipynb @@ -215,7 +215,11 @@ "\n", "# Import the `onnx_utils` function to optimize the model:\n", "# Import the `onnx_utils` function to optimize the model:\n", - "onnx_func = project.set_function(\"hub://onnx_utils\", name=\"onnx-utils\",requirements=['tensorflow==2.9.0'] if framework=='tf-keras' else ['torch'])\n", + "onnx_func = project.set_function(\"hub://onnx_utils\", name=\"onnx-utils\",requirements=['tensorflow==2.9.0' 'onnx~=1.10.1',\n", + " 'onnxruntime~=1.8.1',\n", + " 'onnxoptimizer~=0.2.0',\n", + " 'onnxmltools~=1.9.0',\n", + " 'tf2onnx~=1.9.0'] if framework=='tf-keras' else ['torch'])\n", "\n", "onnx_func.deploy()\n", "\n", From 8ee891dd08187608a7bdab26a1fbc998075bc310 Mon Sep 17 00:00:00 2001 From: Avi Asulin Date: Tue, 31 Oct 2023 14:42:03 +0200 Subject: [PATCH 2/2] [mask-detection] add onnx missing requirements --- mask-detection/3-automatic-pipeline.ipynb | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mask-detection/3-automatic-pipeline.ipynb b/mask-detection/3-automatic-pipeline.ipynb index 6bcbff03..6a957696 100644 --- a/mask-detection/3-automatic-pipeline.ipynb +++ b/mask-detection/3-automatic-pipeline.ipynb @@ -215,7 +215,8 @@ "\n", "# Import the `onnx_utils` function to optimize the model:\n", "# Import the `onnx_utils` function to optimize the model:\n", - "onnx_func = project.set_function(\"hub://onnx_utils\", name=\"onnx-utils\",requirements=['tensorflow==2.9.0' 'onnx~=1.10.1',\n", + "onnx_func = project.set_function(\"hub://onnx_utils\", name=\"onnx-utils\",requirements=['tensorflow==2.9.0',\n", + " 'onnx~=1.10.1',\n", " 'onnxruntime~=1.8.1',\n", " 'onnxoptimizer~=0.2.0',\n", " 'onnxmltools~=1.9.0',\n",