From 1c1db0d2a808287107e15c7d8ed6fd278c0f4023 Mon Sep 17 00:00:00 2001 From: kiritigowda Date: Wed, 6 Mar 2024 10:48:07 -0800 Subject: [PATCH] NN Inference Setup - Use ONNX 1.12.0 --- MIVisionX-setup.py | 2 +- model_compiler/README.md | 2 +- tests/neural_network_tests/runNeuralNetworkTests.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/MIVisionX-setup.py b/MIVisionX-setup.py index 3bc4043546..132d7c66e2 100644 --- a/MIVisionX-setup.py +++ b/MIVisionX-setup.py @@ -349,7 +349,7 @@ # Install CAFFE Deps os.system('sudo pip3 install google==3.0.0 protobuf==3.12.4') # Install ONNX Deps - os.system('sudo pip3 install onnx==1.11.0') + os.system('sudo pip3 install onnx==1.12.0') # Install NNEF Deps os.system('mkdir -p '+modelCompilerDeps+'/nnef-deps') os.system( diff --git a/model_compiler/README.md b/model_compiler/README.md index f7120b7d4a..e169cf8be1 100644 --- a/model_compiler/README.md +++ b/model_compiler/README.md @@ -72,7 +72,7 @@ sudo pip3 install google==3.0.0 protobuf==3.12.4 * protobuf * onnx ``` -sudo pip3 install protobuf==3.12.4 onnx==1.11.0 +sudo pip3 install protobuf==3.12.4 onnx==1.12.0 ``` **Note:** * ONNX Models are available at [ONNX Model Zoo](https://github.com/onnx/models) diff --git a/tests/neural_network_tests/runNeuralNetworkTests.py b/tests/neural_network_tests/runNeuralNetworkTests.py index b9bd120341..ae83543dc3 100644 --- a/tests/neural_network_tests/runNeuralNetworkTests.py +++ b/tests/neural_network_tests/runNeuralNetworkTests.py @@ -279,7 +279,7 @@ def script_info(): # Install CAFFE Deps os.system('sudo pip3 install google==3.0.0 protobuf==3.12.4') # Install ONNX Deps - os.system('sudo pip3 install onnx==1.11.0') + os.system('sudo pip3 install onnx==1.12.0') # Install NNEF Deps os.system('mkdir -p '+modelCompilerDeps+'/nnef-deps') os.system(