tensorflow - 如何将 TensorFlow 版本 1 转换为 onnx 模型？

Question

我试过了python -m tf2onnx.convert --saved-model [file_name] --output [onnx_file_name]。但它是tensorflow = 2.4.4自动运行的。我想运行 tensorflow 版本 1 代码。这段代码可以选择吗？

Answer 1

您可以安装 TensorFlow 版本 1，我也尝试使用也适用于结果的 tf.compat.v1.layers。您可能需要使用 model.save 来获得 .pb 格式并由程序转换。

Answer 2

我用过python -m tf2onnx.convert --saved-model [model file] --output [onnx file name].onnx --opset 13，我解决了。