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"