onnx模型轉tensorflow模型

2021-09-13 01:59:17 字數 922 閱讀 7825

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

1、安裝tensorflow和onnx

我是通過anaconda安裝的。【詳情】這個部落格記載了安裝anaconda和onnx的詳情,安裝好anaconda後,至於安裝tensorflow只需要輸入【conda install tensorflow】就行了。

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

2、安裝onnx-tensorflow

$ git clone [email protected]: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

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

pt模型訓練,轉onnx

pytorch訓練yolov5s.pt,之後轉onnx。以檢測條形碼為例,1分類問題,訓練yolov5s.pt。2 data資料夾裡,images放入樣本,labels放入對應的txt標籤 labelimg製作,注意選擇yolo模式 新建yaml檔案 上圖中的barcode.yaml 用於指定路徑和...

onnx模型轉tensorflow的pb格式

遇到了很多坑,最常見的就是少了什麼模組,或某個模組沒有啥屬性,如 module tensorflow has no attribute ceil 記錄一下現在轉換成功的方式,首先 git clone 之後在終端安裝 pip install e.pip install tensorflow addon...

神經網路模型模型轉ONNX

近期由於業務需要,需要將訓練好的模型轉為onnx格式,為此頗費了一番功夫,在此總結一下吧。1 onnx是一種神經網路模型儲存的中間格式,支援多種格式的模型轉為onnx,也支援使用onnx匯入多種格式的模型,具體見目前其實onnx對於模型的支援還不是太好,主要表現在一些op還不能夠支援 2 在pyto...