Skip to content

Add FallingThings dataset #6346

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 4 commits into from
Aug 17, 2022
Merged

Add FallingThings dataset #6346

merged 4 commits into from
Aug 17, 2022

Conversation

TeodorPoncu
Copy link
Contributor

This is a continuation of the PR split (#6311, #6269) which contains the FallingThings dataset variants.

Copy link
Member

@NicolasHug NicolasHug left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @TeodorPoncu , minor comments from me but LGTM anyway

transforms (callable, optional): A function/transform that takes in a sample and returns a transformed version.
"""

def __init__(self, root: str, variant: str = "single", transforms: Optional[Callable] = None):
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Just trying to figure out whether there are relevant alternatives for "variant" here. Looking at the authors' README I could only find "scene type".

At the root level, there are two folders representing the two types of scenes:

  • single (single falling object), and
  • mixed (2 to 10 falling objects).

Not sure it's better though...

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I've got no particular feelings about this, it's definitely not a "split" therefor I just opted for "variant" to keep consistent with other datasets.

@TeodorPoncu TeodorPoncu merged commit 162267c into main Aug 17, 2022
facebook-github-bot pushed a commit that referenced this pull request Aug 24, 2022
Summary:
* Added Falling Things datasets

* Renamed split to variant

* Update torchvision/datasets/_stereo_matching.py

Changed constant formatting

Reviewed By: datumbox

Differential Revision: D38824218

fbshipit-source-id: adeeeec057bd0afd1fb286e11526b871e62d376f

Co-authored-by: Nicolas Hug <[email protected]>
Co-authored-by: Nicolas Hug <[email protected]>
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