IGEV-Stereo/core/stereo_datasets.py

@@ -73,6 +73,7 @@ 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)
 

IGEV-Stereo/evaluate_stereo.py

@@ -169,7 +169,7 @@ def validate_sceneflow(model, iters=32, mixed_prec=False, max_disp=192):
 
 
 @torch.no_grad()
-def validate_middlebury(model, iters=32, split='H', mixed_prec=False, max_disp=192):
+def validate_middlebury(model, iters=32, split='F', mixed_prec=False, max_disp=192):
     """ Peform validation using the Middlebury-V3 dataset """
     model.eval()
     aug_params = {}