Skip to content

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

Merged
merged 6 commits into from
Sep 9, 2022
Merged

Added Middlebury2014 dataset #6350

merged 6 commits into from
Sep 9, 2022

Conversation

TeodorPoncu
Copy link
Contributor

This is a continuation of the PR split (#6311, #6269) which contains the Middlebury 2014 version dataset.

Copy link
Contributor

@jdsgomes jdsgomes left a 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

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"))
Copy link
Contributor

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):
Copy link
Contributor

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()

@TeodorPoncu TeodorPoncu merged commit a67cc87 into main Sep 9, 2022
facebook-github-bot pushed a commit that referenced this pull request Sep 12, 2022
Summary:
* Added Middlebury2014   dataset

* adressed nits

* mypy fix

Reviewed By: YosuaMichael

Differential Revision: D39426991

fbshipit-source-id: 382a30aee1b4927a5cb47a338f2307ec3241bb71
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants