From e7a1b5978759957aed78c694f06dca09f9ec5c09 Mon Sep 17 00:00:00 2001 From: CptCaptain Date: Tue, 30 Nov 2021 12:56:44 +0100 Subject: [PATCH] fix parenthesis --- data/dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/data/dataset.py b/data/dataset.py index c717498..7e282d0 100644 --- a/data/dataset.py +++ b/data/dataset.py @@ -89,9 +89,9 @@ class TrackSynDataset(torchext.BaseDataset): R = [] t = [] for tidx in track_ind: - disps.append(np.load(os.path.join(sample_path, f'disp0_{tidx}.npy')), allow_pickle=True) - R.append(np.load(os.path.join(sample_path, f'R_{tidx}.npy')), allow_pickle=True) - t.append(np.load(os.path.join(sample_path, f't_{tidx}.npy')), allow_pickle=True) + disps.append(np.load(os.path.join(sample_path, f'disp0_{tidx}.npy'), allow_pickle=True)) + R.append(np.load(os.path.join(sample_path, f'R_{tidx}.npy'), allow_pickle=True)) + t.append(np.load(os.path.join(sample_path, f't_{tidx}.npy'), allow_pickle=True)) ret[f'disp0'] = np.stack(disps, axis=0) ret['R'] = np.stack(R, axis=0) ret['t'] = np.stack(t, axis=0)