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] 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 = {}