使用onnx将Pytorch模型转到tf

安装

按照官方的指导来, 在装好 onnx, pytorch, tensorflow, torchvision 后,

git clone git@github.com:onnx/onnx-tensorflow.git && cd onnx-tensorflow
pip install -e .   #( 建议加上 -i https://pypi.tuna.tsinghua.edu.cn/simple)

使用