Toybrick

标题: TensorFlow生成PB文件出错 [打印本页]

作者: cfan 时间: 2020-12-16 16:19
标题: TensorFlow生成PB文件出错
请问有大佬用tensorflow生成过pb文件吗，rknn-toolkit转换要用到。
怎么能把tensorflow框架训练出来的模型(ckpt)文件转成.pb文件。
教程说找到输出节点，就能把相关的节点都保存到pb文件里，但我保存下来的pb文件只有我指定的那一个节点

作者: leok 时间: 2020-12-18 09:40
找到你要存的节点，再frozen。
多在搜索引擎上搜索一下，并且问问题请提供详细信息；这样没人能给你建议。

欢迎光临 Toybrick (https://t.rock-chips.com/)