added onnx model conversion

fabiocarrara · Feb 3, 2022 · c8dc801 · c8dc801
1 parent f9c3b15
commit c8dc801
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -3,4 +3,4 @@ FROM tensorflow/tensorflow:2.4.1-gpu-jupyter
 RUN apt update && apt install -y nvidia-modprobe
 RUN pip install adabelief-tf pandas imageio imageio-ffmpeg seaborn sklearn tensorflow_addons tqdm
 RUN pip install keras_applications tensorflowjs
-
+RUN pip install tf2onnx
diff --git a/convert_to_onnx.sh b/convert_to_onnx.sh
@@ -0,0 +1,8 @@
+#/bin/bash
+
+# prerequisite:
+# pip install tf2onnx
+
+RUNDIR="path/to/run/dir"
+
+python -m tf2onnx.convert --saved-model $RUNDIR/best_savedmodel --output $RUNDIR/best_model.onnx