​ ValueError: Input 0 of node ... was passed float from ... incompatible with expected float_ref.

在尝试将.pb和.ckpt文件通过freeze_graph.py整合为新的.pb文件时,遇到了错误:ValueError: Input 0 of node... was passed float from... incompatible with expected float_ref。查阅资料后,发现该问题可能源于数据类型不匹配。解决方案通常涉及检查并确保所有节点的数据类型一致。参考链接指向了ONNX与TensorFlow转换中类似问题的讨论。
摘要由CSDN通过智能技术生成

将pb与ckpt通过freeze_graph.py整合生成新的pb文件,对文件执行图导入时,执行

tf.import_graph_def(graph_def, name='')

遇到如下报错:

ValueError: Input 0 of node lanenet_loss/inference/encode/conv1_1/cond/bn/AssignMovingAvg/Switch was passed float from lanenet_loss/inference/encode/conv1_1/bn/moving_mean:0 incompatible with expected float_ref.
原因是在运行freeze_graph.py之后,AssignSub和RefSwitch处理后的输入类型由float_ref变为了float,因此要额外地去转换一下,在图导入之前执行如下代码:
for node in graph_def.node :
	if node.op == 'RefSwitch':
		node.op = 'Switch'
		for index in range(len(node.input)):
			if 'moving_
  • 4
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值