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'])