snapshot...

modules/util/augment.py

@@ -20,6 +20,9 @@ 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%)
             def rotate_fn(low_resolution, high_resolution):