constant_graph = graph_util.convert_variables_to_constants(sess, sess.graph_def, ['*/BiasAdd'])
为自己定义的name scope。
如果使用tf.layers.conv2d实现的卷积操作,要将pb模型的输出结果为该卷积的结果,需要指定为 /BiasAdd
而不是/Conv2D
constant_graph = graph_util.convert_variables_to_constants(sess, sess.graph_def, ['*/BiasAdd'])
为自己定义的name scope。
如果使用tf.layers.conv2d实现的卷积操作,要将pb模型的输出结果为该卷积的结果,需要指定为 /BiasAdd
而不是/Conv2D