Files
ascend-tools/pt2tf/README.md
T
2020-09-23 09:09:49 +08:00

16 lines
303 B
Markdown

#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