小编Cro*_*rKZ的帖子

是否可以用现有图形中的常量替换占位符?

我有一个受过训练的模型的冻结图,它有一个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=[])

python tensorflow

8
推荐指数
2
解决办法
4347
查看次数

标签 统计

python ×1

tensorflow ×1