发新帖

freeze.py导出pb文件时,output_node_names是什么

[复制链接]
50 1

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
我看了文档解释说output_node_names是模型的输出节点,我通过python /tensorflow/python/tools/inspect_checkpoint.py --file_name=model-10000-ckpt

打印出了节点信息,但是我不知道怎么找到我要的那个节点,后来查资料说是在定义网络结构的时候,已经定义好了。

在这方面是小白,目前只会在网上在开源的训练,很多代码都看不懂,求大神帮忙解释下,这个模型里面的SRGAN网络结构的输出节点是什么啊?

https://github.com/brade31919/SR ... master/lib/model.py
我知道答案 回答被采纳将会获得10 金币 + 5 金币 已有1人回答
本楼点评(0) 收起

精彩评论1

ViolinSolo  TF豆豆  发表于 2018-7-13 15:41:42 | 显示全部楼层
我觉得你可以试一下定义一下name,这样子你打印ckpt里面的变量的时候,就好找了,,要不然像我用个resnet50,然后变量多成狗,根本分不清哈哈哈
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

主题

帖子

3

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