From 860ed6deae0b9aaac042c5cb8b7bbdb9173bd395 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 2 Aug 2023 12:54:59 -0500
Subject: [PATCH] snapshot...
---
modules/util/augment.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/modules/util/augment.py b/modules/util/augment.py
index 4a418179..788bd694 100644
--- a/modules/util/augment.py
+++ b/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):
--
GitLab