使用tf.layers.conv2d()转PB模型时记得加Conv2D后缀

tf.layers.conv2d()会自动添加Conv2D

例如我们这样设置

tf.layers.conv2d( , name='output')

我们将checkpoint转换成pb模型时,需要制定一个输出节点。使用layers这个包,如name为output,如果直接设定该节点为输出节点,则会报错。
原因:tensorflow的layers会自动的name的后面加上Conv2D下级scope,所以我们需要这样设置输出节点

from tensorflow.python.framework import graph_util
...
...
pb_graph = graph_util.convert_variables_to_constants(sess, sess.graph_def, ['output/Conv2D'])
发布了49 篇原创文章 · 获赞 14 · 访问量 19万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览