geekyrakshit commited on
Commit
c23af12
1 Parent(s): 1bb0e15

bug fix in augmentation factory

Browse files
Files changed (1) hide show
  1. enhance_me/augmentation.py +1 -1
enhance_me/augmentation.py CHANGED
@@ -44,7 +44,7 @@ class AugmentationFactory:
44
  ),
45
  )
46
 
47
- def random_rotate(input_image, enhanced_image):
48
  condition = tf.random.uniform(shape=(), maxval=4, dtype=tf.int32)
49
  return tf.image.rot90(input_image, condition), tf.image.rot90(
50
  enhanced_image, condition
 
44
  ),
45
  )
46
 
47
+ def random_rotate(self, input_image, enhanced_image):
48
  condition = tf.random.uniform(shape=(), maxval=4, dtype=tf.int32)
49
  return tf.image.rot90(input_image, condition), tf.image.rot90(
50
  enhanced_image, condition