From 091e1963dcfcf2a0d261d4eb2f4e5665d512eda7 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Wed, 6 Sep 2023 15:16:42 -0500 Subject: [PATCH] snapshot... --- modules/GSOC/E2_ESRGAN/lib/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py index 29335c96..034e3bdf 100644 --- a/modules/GSOC/E2_ESRGAN/lib/dataset.py +++ b/modules/GSOC/E2_ESRGAN/lib/dataset.py @@ -180,12 +180,12 @@ class OpdNpyDataset: fname = self.filenames[fi] data = np.load(fname) - refl = data[0, 0, :, :] + refl = data[:, 0, :, :] refl = scale(refl, 'refl_0_65um_nom', mean_std_dct) refl = refl.astype(np.float32) refl_s.append(refl) - opd = data[0, 1, :, :] + opd = data[:, 1, :, :] opd = scale(opd, 'cld_opd_dcomp', mean_std_dct) opd = opd.astype(np.float32) opd_s.append(opd) -- GitLab