tensorfow checkpoint 转pb 模型 conv2d的输出节点名字是BiasAdd

constant_graph = graph_util.convert_variables_to_constants(sess, sess.graph_def, ['*/BiasAdd'])

为自己定义的name scope。
如果使用tf.layers.conv2d实现的卷积操作,要将pb模型的输出结果为该卷积的结果,需要指定为 /BiasAdd
而不是
/Conv2D

发布了49 篇原创文章 · 获赞 14 · 访问量 19万+
展开阅读全文

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

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

分享到微信朋友圈

×

扫一扫,手机浏览