我现在是项目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]
。