add pt2tf tool

2020-09-23 09:09:49 +08:00
parent 7f7b7df65d
commit 18aefa4dd0
407 changed files with 16211 additions and 0 deletions
@@ -0,0 +1,15 @@
+#1 建立虚拟环境
+$ virtualenv .venv
+
+#2 激活虚拟环境
+$ source .venv/bin/activate
+
+#3 安装依赖包
+$ pip install -r requirements.txt
+$ pip install -e onnx-tensorflow
+
+#4 生成onnx模型
+$ python pt2onnx.py
+
+#5 生成pb模型
+$ onnx-tf convert -i efficientnet-b3.onnx -o efficientnet-b3.pb