Cpt.Captain/connecting_the_dots
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix parenthesis

Browse Source
This commit is contained in:
CptCaptain 2021-11-30 12:56:44 +01:00
parent 0d18f71c6d
commit e7a1b59787
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
data/dataset.py
View File

@ -89,9 +89,9 @@ class TrackSynDataset(torchext.BaseDataset):
R = [] R = []
t = [] t = []
for tidx in track_ind: for tidx in track_ind:
disps.append(np.load(os.path.join(sample_path, f'disp0_{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) 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) 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[f'disp0'] = np.stack(disps, axis=0)
ret['R'] = np.stack(R, axis=0) ret['R'] = np.stack(R, axis=0)
ret['t'] = np.stack(t, axis=0) ret['t'] = np.stack(t, axis=0)