snapshot...

modules/util/augment.py

@@ -20,8 +20,6 @@ def augment_image(
 
     def augment_fn(low_resolution, high_resolution, *args, **kwargs):
         # Augmenting data (~ 80%)
-        low_resolution = tf.convert_to_tensor(low_resolution)
-        high_resolution = tf.convert_to_tensor(high_resolution)
 
         def augment_steps_fn(low_resolution, high_resolution):
             # Randomly rotating image (~50%)