Skip to content

Commit debc43a

Browse files
committed
Add a new class for the CaltechBirdInfo2011 dataset
1 parent 6f6d33e commit debc43a

File tree

1 file changed

+15
-2
lines changed

1 file changed

+15
-2
lines changed

tensorflow_datasets/image_classification/caltech_birds.py

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -231,9 +231,9 @@ class CaltechBirds2011(CaltechBirds2010):
231231

232232
@property
233233
def _caltech_birds_info(self):
234-
return CaltechBirdsInfo(
234+
return CaltechBirdsInfo2011(
235235
name=self.name,
236-
combined_url="https://data.caltech.edu/records/65de6-vp158/files/CUB_200_2011.tgz?download=1",
236+
combined_url="https://data.caltech.edu/records/65de6-vp158/files/CUB_200_2011.tgz?download=1"
237237
# images_url="https://drive.google.com/uc?export=download&id=1hbzc_P1FuxMkcabkgn9ZKinBwW683j45",
238238
# split_url=None,
239239
# annotations_url="https://drive.google.com/uc?export=download&id=1EamOKGLoTuZdtcVYbHMWNpkn3iAVj8TP",
@@ -371,3 +371,16 @@ class CaltechBirdsInfo(
371371
annotations_url (str): annotation folder URL.
372372
"""
373373

374+
375+
class CaltechBirdsInfo2011(
376+
collections.namedtuple(
377+
"_CaltechBirdsInfo2011",
378+
["name", "combined_url"],
379+
)
380+
):
381+
"""Contains the information necessary to generate a Caltech Birds 2011 dataset.
382+
383+
Args:
384+
name (str): name of dataset.
385+
combined_url (str): URL containing images and attributes.
386+
"""

0 commit comments

Comments
 (0)