def image_rotation(x):
""" imgs, scale, scale is in radians """
rands = tf.truncated_normal([tf.shape(x)[0]], stddev=0.05)
return images_rotate(x, rands, interpolation='BILINEAR')
def call(self, x):
""" imgs, scale, scale is in radians """
rands = tf.truncated_normal([tf.shape(x[0])[0]], stddev=x[1])
return images_rotate(x[0], rands, interpolation='BILINEAR')
