tf_graph = from_onnx_model(onnx_model) 没有from_onnx_model函数
时间: 2023-10-04 22:04:01 浏览: 83
tensorflow使用freeze_graph.py将ckpt转为pb文件的方法
是的,确实没有 `from_onnx_model` 函数。如果你想将一个 ONNX 模型转换为 Tensorflow 模型,你可以使用 `tf.compat.v1.onnx.parse_from_string` 函数来解析 ONNX 模型,并使用 `tf.compat.v1.graph_util.import_graph_def` 函数将解析出的图导入到 Tensorflow 的默认图中。下面是一个示例代码:
```python
import onnx
# Load the ONNX model
onnx_model = onnx.load("model.onnx")
# Parse the ONNX model and create a Tensorflow graph
tf_graph = tf.compat.v1.onnx.parse_from_string(onnx_model.SerializeToString())
# Import the Tensorflow graph into the default graph
tf.import_graph_def(tf_graph)
```
这个代码片段解析了一个名为 `model.onnx` 的 ONNX 模型,并将其导入到 Tensorflow 的默认图中。你可以在导入之后使用 Tensorflow 的其他工具对该模型进行操作和评估。
阅读全文