OpKernel was registered to support Op 'PyFunc' with these attrs

tensorflow1.14   onnx-tf 1.5.0  onnx 1.5.0

pytorch->onnx->tensorflow ,加载pb模型的时候出现如下错误 

Invalid argument: No OpKernel was registered to support Op 'PyFunc' with these attrs.  Registered devices: [CPU], Registered kernels:
  <no registered kernels>

 [[Node: PyFunc = PyFunc[Tin=[DT_FLOAT, DT_INT32, DT_INT32, DT_INT32, DT_INT32, DT_INT32, DT_STRING], Tout=[DT_FLOAT], token="pyfunc_0"](Relu, PyFunc/input_1, PyFunc/input_2, PyFunc/input_3, PyFunc/input_4, PyFunc/input_5, PyFunc/input_6)]]

 

参考https://github.com/onnx/onnx-tensorflow/issues/167                                                                                                                   

from onnx_tf.backend import prepare
model = onnx.load("duc_se.onnx")
tf_rep = prepare(model, strict=False)

测试alphapose的.pd模型和pytorch的结果是一致的

 

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值