ONNX 到 Tensorflow 2:UnicodeDecodeError
ONNX to Tensorflow 2: UnicodeDecodeError
我在 Ubuntu 18.04.
上使用 Tensorflow 2.5.0
我使用以下代码将 ONNX 模型转换为 Tensorflow 2 .pb 格式:
import onnx
from onnx_tf.backend import prepare
model_path="my_model/model.onnx"
output_path="my_model/model.pb"
onnx_model = onnx.load(model_path)
tf_rep = prepare(onnx_model)
tf_rep.export_graph(output_path)
然后我尝试读取转换后的 .pb 文件:
import tensorflow as tf
model_dir = "my_model"
model = tf.saved_model.load(model_dir)
print(model.summary())
我收到以下错误
UnicodeDecodeError: 'utf-8' codec can't decode byte 0x88 in position 89: 'utf-8' codec can't decode byte 0x88 in position 89: invalid start byte in field: tensorflow.MetaGraphDef.MetaInfoDef.meta_graph_version
我该如何解决这个问题?尝试使用谷歌搜索,但我找到的答案都是针对 TF 1 的,它要么使用 "rb"
模式读取,要么使用 "utf8"
解码,但这两个选项在 TF 2 的 tf.saved_model.load()
.[=16= 中都不可用]
我的问题没有答案,但我设法在具有类似软件设置的不同硬件上测试了代码并且它有效。
所以也许我正在处理的原始硬件有一些损坏的库...
我在 Ubuntu 18.04.
上使用 Tensorflow 2.5.0
我使用以下代码将 ONNX 模型转换为 Tensorflow 2 .pb 格式:
import onnx
from onnx_tf.backend import prepare
model_path="my_model/model.onnx"
output_path="my_model/model.pb"
onnx_model = onnx.load(model_path)
tf_rep = prepare(onnx_model)
tf_rep.export_graph(output_path)
然后我尝试读取转换后的 .pb 文件:
import tensorflow as tf
model_dir = "my_model"
model = tf.saved_model.load(model_dir)
print(model.summary())
我收到以下错误
UnicodeDecodeError: 'utf-8' codec can't decode byte 0x88 in position 89: 'utf-8' codec can't decode byte 0x88 in position 89: invalid start byte in field: tensorflow.MetaGraphDef.MetaInfoDef.meta_graph_version
我该如何解决这个问题?尝试使用谷歌搜索,但我找到的答案都是针对 TF 1 的,它要么使用 "rb"
模式读取,要么使用 "utf8"
解码,但这两个选项在 TF 2 的 tf.saved_model.load()
.[=16= 中都不可用]
我的问题没有答案,但我设法在具有类似软件设置的不同硬件上测试了代码并且它有效。 所以也许我正在处理的原始硬件有一些损坏的库...