From 0359609ca79d9b84cc2c84a9b36ca24389232ab3 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 23 Oct 2023 15:28:57 -0500
Subject: [PATCH] snapshot...
---
modules/util/augment.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/modules/util/augment.py b/modules/util/augment.py
index 9668d780..68c39c61 100644
--- a/modules/util/augment.py
+++ b/modules/util/augment.py
@@ -59,8 +59,8 @@ def augment_image_3arg():
def rotate_fn(data, data_b, label):
times = tf.random.uniform(minval=1, maxval=4, dtype=tf.int32, shape=[])
return (tf.image.rot90(data, times),
- tf.image.rot90(data_b, times),
- tf.image.rot90(label, times))
+ data_b,
+ label)
data, data_b, label = tf.cond(
tf.less_equal(tf.random.uniform([]), 0.5),
@@ -70,8 +70,8 @@ def augment_image_3arg():
# Randomly flipping image (~50%)
def flip_fn(data, data_b, label):
return (tf.image.flip_left_right(data),
- tf.image.flip_left_right(data_b),
- tf.image.flip_left_right(label))
+ data_b,
+ label)
data, data_b, label = tf.cond(
tf.less_equal(tf.random.uniform([]), 0.5),
--
GitLab