Skip to content

Add Flowers102 dataset #5177

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 28 commits into from
Jan 12, 2022
Merged

Add Flowers102 dataset #5177

merged 28 commits into from
Jan 12, 2022

Conversation

zhiqwang
Copy link
Contributor

@zhiqwang zhiqwang commented Jan 8, 2022

Addresses Flowers102 dataset in #5108.

cc @pmeier @NicolasHug

@facebook-github-bot
Copy link

facebook-github-bot commented Jan 8, 2022

💊 CI failures summary and remediations

As of commit 0791dfc (more details on the Dr. CI page):


💚 💚 Looks good so far! There are no failures yet. 💚 💚


This comment was automatically generated by Dr. CI (expand for details).

Please report bugs/suggestions to the (internal) Dr. CI Users group.

Click here to manually regenerate this comment.

Copy link
Collaborator

@pmeier pmeier left a comment

Choose a reason for hiding this comment

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

Hey @zhiqwang, thanks a lot for the PR. I have some minor comments inline.

Copy link
Collaborator

@pmeier pmeier left a comment

Choose a reason for hiding this comment

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

Thanks a lot @zhiqwang! I've pushed a minor simplification to not prolong this. LGTM from my side!

@pmeier pmeier requested a review from NicolasHug January 12, 2022 10:35
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 a lot @zhiqwang for the PR, and @pmeier for the review.

I made some very minor changes above, the only relevant one is that I changed split="valid" into split="val" for more consistency with other datasets.

This LGTM but I just have one question below regarding the download logic, LMK what you think

@zhiqwang zhiqwang requested review from NicolasHug and pmeier January 12, 2022 13:33
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 a lot @zhiqwang !! Will merge when green

@NicolasHug NicolasHug merged commit a8f2ded into pytorch:main Jan 12, 2022
@github-actions
Copy link

Hey @NicolasHug!

You merged this PR, but no labels were added. The list of valid labels is available at https://github.com/pytorch/vision/blob/main/.github/process_commit.py

@zhiqwang zhiqwang deleted the datasets/flowers-102 branch January 12, 2022 15:27
facebook-github-bot pushed a commit that referenced this pull request Jan 17, 2022
Summary:
* Add Flowers102 datasets

* Fix initialization of images and labels

* Fix _check_exists in Flowers102

* Add Flowers102 to datasets and docs

* Add Flowers102TestCase to unittest

* Fixing Python type statically

* Shuffle the fake labels

* Update test/test_datasets.py

* Apply the suggestions by pmeier

* Use check_integrity to check file existence

* Save the labels to base_folder

* Minor fixes

* Using a loop makes this more concise without reducing readability

* Using a loop makes this more concise without reducing readability

* Remove self.labels and self.label_to_index attributes

* minor simplification

* Check the exitence of image folder

* Revert the check

* Check the existence of image folder

* valid -> val

* keep some stuff private

* minor doc arrangements

* remove default FEATURE_TYPES

* Simplify the datasets existence

* check if the image folder exists

* isdir -> is_dir

Reviewed By: NicolasHug

Differential Revision: D33618165

fbshipit-source-id: ecf9e44202f58bf7a5233248d5471d464efd6644

Co-authored-by: Philip Meier <[email protected]>
Co-authored-by: Philip Meier <[email protected]>
Co-authored-by: Philip Meier <[email protected]>
Co-authored-by: Philip Meier <[email protected]>
Co-authored-by: Philip Meier <[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.

4 participants