change resolution of middlebury test images

This commit is contained in:
HTensor 2023-05-02 13:39:36 +08:00
parent e7033dabf9
commit a7d89bd95c

View File

@ -294,7 +294,7 @@ class CREStereo(StereoDataset):
class Middlebury(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) super(Middlebury, self).__init__(aug_params, sparse=True, reader=frame_utils.readDispMiddlebury)
assert os.path.exists(root) assert os.path.exists(root)
assert split in "FHQ" assert split in "FHQ"