Skip to content

Commit a200dac

Browse files
authored
Add classes to the Flowers102 dataset (#8838)
1 parent fab1188 commit a200dac

File tree

1 file changed

+105
-0
lines changed

1 file changed

+105
-0
lines changed

torchvision/datasets/flowers102.py

Lines changed: 105 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,3 +112,108 @@ def download(self):
112112
for id in ["label", "setid"]:
113113
filename, md5 = self._file_dict[id]
114114
download_url(self._download_url_prefix + filename, str(self._base_folder), md5=md5)
115+
116+
classes = [
117+
"pink primrose",
118+
"hard-leaved pocket orchid",
119+
"canterbury bells",
120+
"sweet pea",
121+
"english marigold",
122+
"tiger lily",
123+
"moon orchid",
124+
"bird of paradise",
125+
"monkshood",
126+
"globe thistle",
127+
"snapdragon",
128+
"colt's foot",
129+
"king protea",
130+
"spear thistle",
131+
"yellow iris",
132+
"globe-flower",
133+
"purple coneflower",
134+
"peruvian lily",
135+
"balloon flower",
136+
"giant white arum lily",
137+
"fire lily",
138+
"pincushion flower",
139+
"fritillary",
140+
"red ginger",
141+
"grape hyacinth",
142+
"corn poppy",
143+
"prince of wales feathers",
144+
"stemless gentian",
145+
"artichoke",
146+
"sweet william",
147+
"carnation",
148+
"garden phlox",
149+
"love in the mist",
150+
"mexican aster",
151+
"alpine sea holly",
152+
"ruby-lipped cattleya",
153+
"cape flower",
154+
"great masterwort",
155+
"siam tulip",
156+
"lenten rose",
157+
"barbeton daisy",
158+
"daffodil",
159+
"sword lily",
160+
"poinsettia",
161+
"bolero deep blue",
162+
"wallflower",
163+
"marigold",
164+
"buttercup",
165+
"oxeye daisy",
166+
"common dandelion",
167+
"petunia",
168+
"wild pansy",
169+
"primula",
170+
"sunflower",
171+
"pelargonium",
172+
"bishop of llandaff",
173+
"gaura",
174+
"geranium",
175+
"orange dahlia",
176+
"pink-yellow dahlia?",
177+
"cautleya spicata",
178+
"japanese anemone",
179+
"black-eyed susan",
180+
"silverbush",
181+
"californian poppy",
182+
"osteospermum",
183+
"spring crocus",
184+
"bearded iris",
185+
"windflower",
186+
"tree poppy",
187+
"gazania",
188+
"azalea",
189+
"water lily",
190+
"rose",
191+
"thorn apple",
192+
"morning glory",
193+
"passion flower",
194+
"lotus",
195+
"toad lily",
196+
"anthurium",
197+
"frangipani",
198+
"clematis",
199+
"hibiscus",
200+
"columbine",
201+
"desert-rose",
202+
"tree mallow",
203+
"magnolia",
204+
"cyclamen",
205+
"watercress",
206+
"canna lily",
207+
"hippeastrum",
208+
"bee balm",
209+
"ball moss",
210+
"foxglove",
211+
"bougainvillea",
212+
"camellia",
213+
"mallow",
214+
"mexican petunia",
215+
"bromelia",
216+
"blanket flower",
217+
"trumpet creeper",
218+
"blackberry lily",
219+
]

0 commit comments

Comments
 (0)