我有一个受过训练的模型的冻结图,它有一个tf.placeholder我总是提供相同的值.
我想知道是否有可能替换它tf.constant.如果它是某种方式 - 任何例子将不胜感激!
编辑:以下是代码的外观,以帮助可视化问题
我正在使用预先训练的(由其他人)模型进行推理.模型在本地存储为具有.pb扩展名的冻结图形文件.
代码如下所示:
# load graph
graph = load_graph('frozen.pb')
session = tf.Session(graph=graph)
# Get input and output tensors
images_placeholder = graph.get_tensor_by_name("input:0")
output = graph.get_tensor_by_name("output:0")
phase_train_placeholder = graph.get_tensor_by_name("phase_train:0")
feed_dict = {images_placeholder: images, phase_train_placeholder: False}
result = session.run(output, feed_dict=feed_dict)
Run Code Online (Sandbox Code Playgroud)
问题是我总是phase_train_placeholder: False以我的目的为食,所以我想知道是否有可能消除占位符并用类似的东西替换它tf.constant(False, dtype=bool, shape=[])