From 9df896bb706828a196f4f6658c6150137a3e5e13 Mon Sep 17 00:00:00 2001 From: HTensor <27774381+HTensor@users.noreply.github.com> Date: Mon, 1 May 2023 12:25:21 +0800 Subject: [PATCH] Update stereo_datasets.py --- IGEV-Stereo/core/stereo_datasets.py | 1 - 1 file changed, 1 deletion(-) diff --git a/IGEV-Stereo/core/stereo_datasets.py b/IGEV-Stereo/core/stereo_datasets.py index 21fc63f..20a4e83 100644 --- 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)