簡體   English   中英

如何將 Tensorflow 版本 1 轉換為 onnx model?

[英]How could I convert Tensorflow version 1 to onnx model?

我試過python -m tf2onnx.convert --saved-model [file_name] --output [onnx_file_name] 但它由tensorflow = 2.4.4自動運行。 我想運行 tensorflow 版本 1 代碼。 這段代碼可以選擇嗎?

您可以安裝 TensorFlow 版本 1,我也嘗試使用也適用於結果的 tf.compat.v1.layers。 您可能需要使用 model.save 來生成 .pb 格式並由程序轉換。

保存並遵守 執行 示例項目

我使用python -m tf2onnx.convert --saved-model [model file] --output [onnx file name].onnx --opset 13我解決了它。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM