From 0ef919df04aa92b13b5bdcf0d857f5a0fb58e2d9 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 6 Sep 2023 14:45:05 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py
index 29335c96..3ec04b42 100644
--- a/modules/GSOC/E2_ESRGAN/lib/dataset.py
+++ b/modules/GSOC/E2_ESRGAN/lib/dataset.py
@@ -193,7 +193,7 @@ class OpdNpyDataset:
opd = np.concatenate(opd_s)
refl = np.concatenate(refl_s)
- hr_image = np.stack([refl, opd], axis=3)
+ hr_image = np.stack([refl, opd], axis=2)
hr_image = tf.image.crop_to_bounding_box(hr_image, 0, 0, self.hr_size, self.hr_size)
low_resolution = tf.image.resize(hr_image, [self.lr_size, self.lr_size], method='bicubic')
--
GitLab