Fix incorrect parenthesis

master
CptCaptain 3 years ago
parent 9325503b51
commit e0450a57b9
  1. 6
      data/dataset.py

@ -77,9 +77,9 @@ class TrackSynDataset(torchext.BaseDataset):
ambs = [] ambs = []
grads = [] grads = []
for tidx in track_ind: for tidx in track_ind:
imgs.append(np.load(os.path.join(sample_path, f'im{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) 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) 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'im{sidx}'] = np.stack(imgs, axis=0)
ret[f'ambient{sidx}'] = np.stack(ambs, axis=0) ret[f'ambient{sidx}'] = np.stack(ambs, axis=0)
ret[f'grad{sidx}'] = np.stack(grads, axis=0) ret[f'grad{sidx}'] = np.stack(grads, axis=0)

Loading…
Cancel
Save