Toybrick

TensorFlow生成PB文件出错

cfan

注册会员

积分
193
楼主
发表于 2020-12-16 16:19:24    查看: 8196|回复: 1 | [复制链接]    打印 | 只看该作者
请问有大佬用tensorflow生成过pb文件吗,rknn-toolkit转换要用到。
怎么能把tensorflow框架训练出来的模型(ckpt)文件转成.pb文件。
教程说找到输出节点,就能把相关的节点都保存到pb文件里,但我保存下来的pb文件只有我指定的那一个节点

回复

使用道具 举报

leok

版主

积分
894
沙发
发表于 2020-12-18 09:40:18 | 只看该作者
找到你要存的节点,再frozen。
多在搜索引擎上搜索一下,并且问问题请提供详细信息;这样没人能给你建议。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表