keras Onnx模型保存问题“元组”对象没有属性“图形”

goqiplq2  于 2023-02-23  发布在  其他
关注(0)|答案(1)|浏览(195)

在将keras模型更改为onnx的过程中,我发现错误“'tuple'对象没有属性'graph'”通过使用Onnx我的代码如下,请帮助我

import tf2onnx
import onnx
from tensorflow.python.keras import models

model = models.load_model("model.h5")   

onnx_model = tf2onnx.convert.from_keras(model, opset=13)

onnx.save(model, 'test.onnx')

“C:\用户\测试\PycharmProjects\OnnxTest\main.py”,第13行,在onnx.save中(onnx_model,'test. onnx')文件“C:\用户\大图片\anaconda3\库\站点包\onnx_init_.py”,第195行,在保存_model中proto =写入_外部_数据_Tensor(原型,基础路径)文件“C:\用户\测试\anaconda3\库\站点包\onnx\外部_数据_助手.py”,第276行,在写入_外部_数据_Tensor中,用于Tensorin _get_all_tensors(模型):文件“C:\用户\测试\anaconda3\库\站点包\onnx\外部数据助手.py”,第207行,in _get_initializer_tensors yield from _get_initializer_tensors_from_graph(onnx_model_proto.graph)属性错误:'tuple'对象没有属性'graph'

3pmvbmvn

3pmvbmvn1#

tf2onnx.convert.from_keras返回2个值。您只接受1个返回值,因此该方法将这两个值作为单个元组返回。添加逗号和下划线以接收和忽略额外返回值的这个小更改将解决此问题:
onx,_= tf2onnx.convert.from_角(模型)
更多信息:https://bobbyhadz.com/blog/python-return-multiple-values-use-only-one

相关问题