HTensor/IGEV
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update stereo_datasets.py

Browse Source
This commit is contained in:
HTensor 2023-05-01 12:25:21 +08:00
parent 73e65f99b8
commit 9df896bb70
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
IGEV-Stereo/core/stereo_datasets.py
View File

@ -73,7 +73,6 @@ class StereoDataset(data.Dataset):
img2 = np.array(img2).astype(np.uint8) img2 = np.array(img2).astype(np.uint8)
disp = np.array(disp).astype(np.float32) disp = np.array(disp).astype(np.float32)
assert not (True in np.isnan(disp))
flow = np.stack([disp, np.zeros_like(disp)], axis=-1) flow = np.stack([disp, np.zeros_like(disp)], axis=-1)