-
Notifications
You must be signed in to change notification settings - Fork 7.1k
Added Middlebury2014 dataset #6350
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, just left some nits
test/test_datasets.py
Outdated
datasets_utils.create_image_file(root=scene_dir, name="im1E.png", size=(3, 100, 100)) | ||
datasets_utils.create_image_file(root=scene_dir, name="im1L.png", size=(3, 100, 100)) | ||
# these are going to end up being gray scale images | ||
datasets_utils.make_fake_pfm_file(h=100, w=100, file_name=os.path.join(scene_dir, "disp0.pfm")) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nit: You can use scene_dir / fname here and in the line bellow to be more consistent with the rest of the PR
scene_url = f"{base_url}/{scene_name}.zip" | ||
print(f"Downloading {scene_url}") | ||
# download the scene only if it doesn't exist | ||
if not os.path.exists(split_root / scene_name): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Another nit: you can just use exists from pathlib: (solit_root / scene_name).exists()
Summary: * Added Middlebury2014 dataset * adressed nits * mypy fix Reviewed By: YosuaMichael Differential Revision: D39426991 fbshipit-source-id: 382a30aee1b4927a5cb47a338f2307ec3241bb71
This is a continuation of the PR split (#6311, #6269) which contains the Middlebury 2014 version dataset.