我在python中有一个基本的tensorflow模型，我想把它转换成onnx-fi

Question

我有python tensorflow的基本模型，我想把它保存到onnx文件中，我该怎么做呢。我试过使用onnx.save函数，我得到了错误。在

File "tenserflowbase.py", line 21, in <module> onnx.save(trained_model,'model.onxx') 
File "C:\Users\Parag_IK\Anaconda3\lib\site-packages\onnx\__init__.py", line 184, in save_model proto = write_external_data_tensors(proto, basepath)                                                                                                                                                                                                                                                           
File "C:\Users\Parag_IK\Anaconda3\lib\site packages\onnx\external_data_helper.py", line 225, in write_external_data_tensors                                               
for tensor in _get_all_tensors(model):                                                                                                                              
File "C:\Users\Parag_IK\Anaconda3\lib\site packages\onnx\external_data_helper.py", line 170, in _get_initializer_tensors                                                 
for initializer in onnx_model_proto.graph.initializer:                                                                                                              
AttributeError: 'History' object has no attribute 'graph'** 

我的代码如下：

^{pr2}$

Answer 1

我想如果我没有错，为了使用onnx.save()，模型应该在onnx函数中创建图。在

所以我建议使用tf2onnx库，它有一个将tf会话图转换为onnx图的函数。在

onnx_graph = tf2onnx.tfonnx.process_tf_graph(sess.graph, ...) 

例如，完整的代码应该是：

^{pr2}$

希望这有帮助。在

参考号：tensorflow-onnx