python - tensorflow 通过随机因素调整图像大小

Question

当图像大小未知时，我试图在图形构建期间按一个因子调整图像大小:

H, W, _ = img.get_shape()
scale = tf.random_uniform([1], minval=1, maxval=1.5, dtype=tf.float32, seed=None, name=None)

在这些中，我需要神奇地计算出一个可以转换为以下内容的大小:

tf.image.resize_images(tf.expand_dims(img, 0), [H*scale, W*scale])

哪个返回:

ValueError: 'size' must be a 1-D int32 Tensor

非常感谢任何帮助。谢谢。

Answer 1

scale 的形状为 (1,)，因此 [H*scale, W*scale] 的形状为 (1 , 2)。要修复它，只需生成一个标量随机数而不是一个 1 元素向量:

scale = tf.random_uniform([], minval=1, maxval=1.5, dtype=tf.float32, seed=None, name=None)

请注意形状是一个空列表，这意味着您需要一个标量。

除此之外，您还必须修复 size 参数的数据类型，如下所示:

tf.image.resize_images(tf.expand_dims(img, 0), tf.cast([H*scale, W*scale], tf.int32))