Update stereo_datasets.py

2023-05-01 12:25:21 +08:00 · 2023-05-01 12:25:21 +08:00 · 9df896bb70
commit 9df896bb70
parent 73e65f99b8
1 changed files with 0 additions and 1 deletions
--- a/IGEV-Stereo/core/stereo_datasets.py
+++ b/IGEV-Stereo/core/stereo_datasets.py
@ -73,7 +73,6 @@ class StereoDataset(data.Dataset):
        img2 = np.array(img2).astype(np.uint8)

        disp = np.array(disp).astype(np.float32)
-        assert not (True in np.isnan(disp))

        flow = np.stack([disp, np.zeros_like(disp)], axis=-1)