From e0450a57b9dfb4164faee2c4a998393db14ae93a Mon Sep 17 00:00:00 2001 From: CptCaptain Date: Wed, 8 Dec 2021 09:47:42 +0100 Subject: [PATCH] Fix incorrect parenthesis --- data/dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/data/dataset.py b/data/dataset.py index 7e282d0..fe7d359 100644 --- a/data/dataset.py +++ b/data/dataset.py @@ -77,9 +77,9 @@ class TrackSynDataset(torchext.BaseDataset): ambs = [] grads = [] for tidx in track_ind: - imgs.append(np.load(os.path.join(sample_path, f'im{sidx}_{tidx}.npy')), allow_pickle=True) - ambs.append(np.load(os.path.join(sample_path, f'ambient{sidx}_{tidx}.npy')), allow_pickle=True) - grads.append(np.load(os.path.join(sample_path, f'grad{sidx}_{tidx}.npy')), allow_pickle=True) + imgs.append(np.load(os.path.join(sample_path, f'im{sidx}_{tidx}.npy'), allow_pickle=True)) + ambs.append(np.load(os.path.join(sample_path, f'ambient{sidx}_{tidx}.npy'), allow_pickle=True)) + grads.append(np.load(os.path.join(sample_path, f'grad{sidx}_{tidx}.npy'), allow_pickle=True)) ret[f'im{sidx}'] = np.stack(imgs, axis=0) ret[f'ambient{sidx}'] = np.stack(ambs, axis=0) ret[f'grad{sidx}'] = np.stack(grads, axis=0)