是否可以从冻结图中删除批次尺寸?

格洛姆

检查冻结的张量流模型:

wget https://storage.googleapis.com/download.tensorflow.org/models/inception_v3_2016_08_28_frozen.pb.tar.gz

我看到输入大小为Tensor 'input:0', which has shape '(1, 299, 299, 3)',我想知道是否可以(None, 299, 299, 3)通过batch_size> 1进行输入以使批次预测可用?

Jdehesa

在一般情况下,可能无法执行此操作,因为可能存在依赖于第一维为1的操作(例如假设tf.squeeze用于input:0)。但是,您可以尝试用所需形状的占位符替换输入。您可以使用tf.graph_util.import_graph_def如果操作允许,则TensorFlow应该导入图以相应地调整节点形状。请参见以下示例:

import tensorflow as tf

# First graph
with tf.Graph().as_default():
    x = tf.placeholder(tf.float32, [1, 10, 20], name='Input')
    y = tf.square(x, name='Output')
    print(y)
    # Tensor("Output:0", shape=(1, 10, 20), dtype=float32)
    gd = tf.get_default_graph().as_graph_def()

# Second graph
with tf.Graph().as_default():
    x = tf.placeholder(tf.float32, [None, 10, 20], name='Input')
    y, = tf.graph_util.import_graph_def(gd, input_map={'Input:0': x},
                                        return_elements=['Output:0'], name='')
    print(y)
    # Tensor("Output:0", shape=(?, 10, 20), dtype=float32)

在第一个图中,Output:0节点的形状(1, 10, 20)是从Input:0张量的形状推断的但是,当我从第一个图形中获取图形定义并在第二个图形中加载时,将Input:0张量替换为具有未定义第一维的占位符,则的形状将Output:0更新为(?, 10, 20)如果我在第二个图中运行的操作给出的输入值的第一维大于一个维度,则它将运行正常,因为该图是正确的。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

是否可以删除Kendo饼图中的“零”标签?

是否可以从Android谷歌地图中删除默认的兴趣点

是否可以从Webpack源地图中删除“ sourcesContent”?

cpp代码主线程因删除对象而冻结是否可以使用智能指针解决?

是否可以删除一个元素的类时,改变屏幕尺寸?

NCO:是否可以在不修改坐标属性的情况下删除尺寸?

是否可以基于批次标签(y_true)分布来更新每个批次的学习率?

是否可以在 git 中命名不同批次的暂存文件?

是否可以禁用(冻结)程序包?

是否可以从Ruby中的String#split函数返回冻结的重复数据删除字符串?

是否可以在自定尺寸单元中使用自定尺寸单元?

是否可以删除vi?

是否可以删除`:〜`目录

是否可以删除信号

TensorFlow:是否运行的每个会话都会在图中启动不同的数据批次?

使用最少的批次发送不同尺寸的文档

临时合并Keras中的批次尺寸

问题:机架被占用。是否可以从机架系统中删除代理而不在流程图中使用它们?

是否可以在输入视图中显示布局?

是否可以颠倒列表视图中的顺序?

是否可以用花括号创建冻结集

是否可以防止从冻结的python数据类读取?

是否可以在Spark批次上创建模型并将其用于Spark流中?

是否可以根据时间边界而不是记录数在 Apache Arrow 中定义记录批次?

批次的括号是否可以容纳多少输出量有限制?

春季批次中的一个步骤是否可以有多个阅读器?

是否可以删除“检查元素”?

是否可以删除或覆盖cmdlet?

i3是否可以硬设置全屏尺寸?