|
|
|
|
@ -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) |
|
|
|
|
|
