HTensor/IGEV
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update evaluate_stereo.py

Browse Source
This commit is contained in:
HTensor 2023-04-30 16:03:58 +08:00
parent 0a20c0a001
commit 73e65f99b8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
IGEV-Stereo/evaluate_stereo.py
View File

@ -169,7 +169,7 @@ def validate_sceneflow(model, iters=32, mixed_prec=False):
@torch.no_grad() @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 """ """ Peform validation using the Middlebury-V3 dataset """
model.eval() model.eval()
aug_params = {} aug_params = {}