ONNX是開源神經網絡交換平臺，有了它基本上不用糾結用什么深度學習框架的問題了。我現在記錄一下怎么將onnx模型轉換成tensorflow模型。

1、安裝tensorflow和onnx

我是通過Anaconda安裝的?！驹斍椤窟@個博客記載了安裝Anaconda和onnx的詳情，安裝好Anaconda后，至于安裝tensorflow只需要輸入【conda install tensorflow】就行了。

注意：如果已經通過Anaconda安裝過其他深度模型框架，建議再創建一個新環境，否則安裝tensorflow時，其他深度學習框架將被刪除。反正我原先安裝的caffe2以及onnx，在安裝tensorflow后都就不見了。

2、安裝onnx-tensorflow

$ git clone git@github.com:onnx/onnx-tensorflow.git && cd onnx-tensorflow ###通過git安裝onnx-tensorflow模塊 $ pip install -e . ###注意那個點也是需要的

通過git進行拉取時，你可能會遇到permission錯誤，這個不是能通過sudo解決的。因為這個錯誤的原因不在這，欲知詳情請到這個博客中查看。

安裝好后，試一下【onnx-tf -h】這個命令可不可以用。沒報錯就行了

3、onnx轉tensorflow

從 ONNX 到?Tensorflow:?onnx-tf convert -t tf -i /path/to/input.onnx -o /path/to/output.pb

反正現在看起來是轉成功了。哈哈哈哈哈哈

?

總結

以上是生活随笔為你收集整理的onnx模型转tensorflow模型的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。