tensorboard ckpt pb 模型的输出节点_使用tf的c++API加载pb做前向推理

e3d6ec8fe1713e61f0677396806379fd.png

前面我们介绍了将tensorflow的ckpt模型文件转换为pb文件,接下来,我们使用tensorflow的C++接口调用pb文件做前向推理。

这里假设已经编译好了tensorflow的动态库

加载模型

加载模型的过程和python接口时一致,先创建一个session会话,定义一个graph,读取pb文件后使用session为定义的graph分配资源。

using 

前向推理

前向推理时,需要指定起止结点,根据指定的起止结点去运行要运行哪部分graph,还要给起止结点的tensor,为输入图像和输出结果开辟内存,以上篇转换pb时定义的tensor name为例。

int 

封装

将加载模型和前向推理封装,伪代码如下:

int 

这样,就可以使用c++代码进行tensorflow的前向推理了。

130171b01730370af50ab95796c3f534.png
长按关注微信公众号
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值