张量大小错误

时间:2017-03-11 08:23:18

标签: python tensorflow deep-learning tensorflow-serving

我现在是项目deeplab_v2,它至少得到了tf r0.11的支持,但我的实验室机器上只有r0.10。因此,当我更改某些代码以适应旧版本时,我会遇到以下问题。

 if input_size is not None:
    h, w = input_size
    if random_scale:
        scale = tf.random_uniform([1], minval=0.75, maxval=1.25, dtype=tf.float32, seed=None)
        h_new = tf.to_int32(tf.mul(tf.to_float(tf.shape(img)[1]), scale))
        w_new = tf.to_int32(tf.mul(tf.to_float(tf.shape(img)[1]), scale))
        new_shape = tf.squeeze(tf.pack([h_new, w_new]), squeeze_dims=[1])

        img = tf.image.resize_images(img, new_shape)

这是原始代码,input_size是[312,312],而tf.image.resize_image(img, [height, width])是tf r0.11中的函数。

但是在tf r0.10中,该函数已更改为tf.image.size_image(img, height, width),因此我不知道如何将new_shape更改为两个元素[height , width]

0 个答案:

没有答案