From a7d89bd95c4dc0bac5de941d2d1945bfc84d5d89 Mon Sep 17 00:00:00 2001 From: HTensor <27774381+HTensor@users.noreply.github.com> Date: Tue, 2 May 2023 13:39:36 +0800 Subject: [PATCH] change resolution of middlebury test images --- IGEV-Stereo/core/stereo_datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/IGEV-Stereo/core/stereo_datasets.py b/IGEV-Stereo/core/stereo_datasets.py index 21fc63f..c8a0835 100644 --- a/IGEV-Stereo/core/stereo_datasets.py +++ b/IGEV-Stereo/core/stereo_datasets.py @@ -294,7 +294,7 @@ class CREStereo(StereoDataset): class Middlebury(StereoDataset): - def __init__(self, aug_params=None, root='/data/Middlebury', split='F'): + def __init__(self, aug_params=None, root='/data/Middlebury', split='H'): super(Middlebury, self).__init__(aug_params, sparse=True, reader=frame_utils.readDispMiddlebury) assert os.path.exists(root) assert split in "FHQ"