From 73e65f99b80deadd8eadace93c20029262d3f91b Mon Sep 17 00:00:00 2001 From: HTensor <27774381+HTensor@users.noreply.github.com> Date: Sun, 30 Apr 2023 16:03:58 +0800 Subject: [PATCH 1/2] Update evaluate_stereo.py --- IGEV-Stereo/evaluate_stereo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/IGEV-Stereo/evaluate_stereo.py b/IGEV-Stereo/evaluate_stereo.py index 73d35dd..e7ff54a 100644 --- a/IGEV-Stereo/evaluate_stereo.py +++ b/IGEV-Stereo/evaluate_stereo.py @@ -169,7 +169,7 @@ def validate_sceneflow(model, iters=32, mixed_prec=False): @torch.no_grad() -def validate_middlebury(model, iters=32, split='F', mixed_prec=False): +def validate_middlebury(model, iters=32, split='H', mixed_prec=False): """ Peform validation using the Middlebury-V3 dataset """ model.eval() aug_params = {} 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 2/2] 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)