-
Notifications
You must be signed in to change notification settings - Fork 7.1k
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
Add Flowers102 dataset #5177
Conversation
💊 CI failures summary and remediationsAs 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. |
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.
Hey @zhiqwang, thanks a lot for the PR. I have some minor comments inline.
Co-authored-by: Philip Meier <[email protected]>
Co-authored-by: Philip Meier <[email protected]>
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.
Thanks a lot @zhiqwang! I've pushed a minor simplification to not prolong this. LGTM from my side!
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.
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
Co-authored-by: Philip Meier <[email protected]>
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.
Thanks a lot @zhiqwang !! Will merge when green
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 |
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]>
Addresses Flowers102 dataset in #5108.
cc @pmeier @NicolasHug