tensorflow 运行下面代码出现问题:
from tensorflow.contrib.image import rotate
......
angle = tf.random_uniform(shape=(1,), minval=-30, maxval=30)
image = rotate(image, angle)
解决:改为在cpu中运行
with tf.device('/cpu:0'):
angle = tf.random_uniform(shape=(1,), minval=-30, maxval=30)
image = rotate(image, angle)