From 7388a6dcf683277cf0bc6ccbf9c9171705a0ae47 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 6 Sep 2023 14:30:26 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/dataset.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py
index be07dd41..29335c96 100644
--- a/modules/GSOC/E2_ESRGAN/lib/dataset.py
+++ b/modules/GSOC/E2_ESRGAN/lib/dataset.py
@@ -136,6 +136,7 @@ f.close()
mean_std_dct.update(mean_std_dct_l1b)
mean_std_dct.update(mean_std_dct_l2)
+
class OpdNpyDataset:
def __init__(self, filenames, hr_size, lr_size, batch_size=128):
@@ -179,12 +180,12 @@ class OpdNpyDataset:
fname = self.filenames[fi]
data = np.load(fname)
- refl = data[0, ]
+ refl = data[0, 0, :, :]
refl = scale(refl, 'refl_0_65um_nom', mean_std_dct)
refl = refl.astype(np.float32)
refl_s.append(refl)
- opd = data[1, ]
+ opd = data[0, 1, :, :]
opd = scale(opd, 'cld_opd_dcomp', mean_std_dct)
opd = opd.astype(np.float32)
opd_s.append(opd)
--
GitLab