onnx模型轉tensorflow的pb格式

2021-10-23 20:56:41 字數 1267 閱讀 9976

遇到了很多坑，最常見的就是少了什麼模組，或某個模組沒有啥屬性，

如: module 『tensorflow』 has no attribute 『ceil』

記錄一下現在轉換成功的方式，

首先：git clone

之後在終端安裝：

pip install -e.


pip install tensorflow-addons
都安裝好了後就可以直接在終端轉換了：

onnx-tf convert -i /path/to/input.onnx -o /path/to/output.pb
轉換失敗大多是版本的原因，

以下我的執行時的版本資訊：

python 3.6.9

onnx version: 1.6.0

onnx-tf version: 1.6.0

tensorflow version: 2.2.0

注意：onnx_tf 無法用pip直接安裝1.6.0的版本，需要從github上原始碼處安裝,

另附上用**轉換的方式：

import onnx


from onnx_tf.backend import prepare


import os


defonnx2pb


(onnx_input_path, pb_output_path)


:    onnx_model = onnx.load(onnx_input_path)


# load onnx model


tf_exp = prepare(onnx_model)


# prepare tf representation


tf_exp.export_graph(pb_output_path)


# export the model


if __name__ ==


"__main__"


:    os.makedirs(


"tensorflow"


, exist_ok=


true


)    onnx_input_path =


'./onnx/facedetectcnn_640.onnx'


pb_output_path =


'./tensorflow/model.pb'


onnx2pb(onnx_input_path, pb_output_path)

onnx模型轉tensorflow模型

onnx是開源神經網路交換平台，有了它基本上不用糾結用什麼深度學習框架的問題了。我現在記錄一下怎麼將onnx模型轉換成tensorflow模型。1 安裝tensorflow和onnx 我是通過anaconda安裝的。詳情 這個部落格記載了安裝anaconda和onnx的詳情，安裝好anaconda後...

pt模型訓練，轉onnx

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

神經網路模型模型轉ONNX

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