File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -33,9 +33,11 @@ mkdir -p model_repository/inception_onnx/1
3333wget -O /tmp/inception_v3_2016_08_28_frozen.pb.tar.gz \
3434 https://storage.googleapis.com/download.tensorflow.org/models/inception_v3_2016_08_28_frozen.pb.tar.gz
3535(cd /tmp && tar xzf inception_v3_2016_08_28_frozen.pb.tar.gz)
36+ apt update -qq && apt install -y python3-venv
37+ rm -rf tf2onnx
3638python3 -m venv tf2onnx
3739source ./tf2onnx/bin/activate
38- pip3 install " numpy<2" tensorflow tf2onnx
40+ pip3 install " numpy<2" tensorflow==2.18.1 tf2onnx==1.16.1 onnx==1.16.1
3941python3 -m tf2onnx.convert --graphdef /tmp/inception_v3_2016_08_28_frozen.pb --output inception_v3_onnx.model.onnx --inputs input:0 --outputs InceptionV3/Predictions/Softmax:0
4042deactivate
4143mv inception_v3_onnx.model.onnx model_repository/inception_onnx/1/model.onnx
You can’t perform that action at this time.
0 commit comments