Question

我是使用tensorflow和python的新手，并且稍微使用它并且无法弄清tf.contrib.image.rotate如何工作，我可以将其用作tensor，例如tf.cast操作。

也许有人可以帮助我弄清楚如何使用它？

格尔茨

Answer 1

您正在混合两个单独的tensorflow函数。

tf.contrib.image.rotate ：此功能用于以您选择的任何程度旋转图像。

X_img =  # Your image or batch of images
degree_angle = 45 # In degrees
radian = degree_angle * math.pi / 180
tf_img = tf.contrib.image.rotate(X_img, radian)
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    rotated_img = sess.run(tf_img)

tf.cast ：用于将张量的dtype从一种形式转换为另一种形式。转换为float32格式的示例。

casted_tensor = tf.cast(input_tensor, tf.float32)

Answer 2

对于那些仍然在这里寻找的人，从TensorFlow 2.x开始tf.contrib已过时，因此tf.contrib.image.rotate将不再起作用。

相反，您可以使用tfa.image.rotate库中的 tensorflow_addons 。此功能逆时针旋转图像，并以弧度作为输入角度。

img = ... #insert image(s) here
angle = 60 #in degrees
angle_radians = angle * math.pi / 180
rotated_img = tfa.image.rotate(img, angle_radians)

tensorflow tf.contrib.image.rotate如何工作？

2 个答案: