Add classes to the Flowers102 dataset#8838
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/8838
Note: Links to docs will display an error until the docs builds have been completed. This comment was automatically generated by Dr. CI and updates every 15 minutes. |
NicolasHug
left a comment
There was a problem hiding this comment.
Thansk for the PR @ageron , this looks good. Just to make sure these are correct, can you share how you created the classes list?
|
Hi Nicolas, So I looked for other sources and found several:
They all agree. I checked that the label counts match, and they all do (adding the training set, validation set, and test set). However, some counts have duplicates: So it's possible that some of these are inverted. That said, I spot checked a few and they looked like the flowers on the Wikipedia, so I'm pretty confident about these labels. |
|
Hi there, does this PR need any tweak or can it be merged? |
|
Thanks @ageron , nothing more to do on your side. I'll merge it eventually, when I have more time to dedicate to torchvision. |
Reviewed By: scotts Differential Revision: D77997075 fbshipit-source-id: 70a5da9e3d198f94df88060dee1e8f8c107579e9
Many datasets provide class names, why not Flowers102?
This PR adds the class names to the
Flowers102class.