Spaces:
Runtime error
Runtime error
geekyrakshit
commited on
Commit
•
c23af12
1
Parent(s):
1bb0e15
bug fix in augmentation factory
Browse files
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
|