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