Skip to content

Commit c8f76f5

Browse files
committed
Add segmentations url and update CaltechBirdsInfo2011 class
1 parent f64b449 commit c8f76f5

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tensorflow_datasets/image_classification/caltech_birds.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,8 @@ class CaltechBirds2011(CaltechBirds2010):
233233
def _caltech_birds_info(self):
234234
return CaltechBirdsInfo2011(
235235
name=self.name,
236-
combined_url="https://data.caltech.edu/records/65de6-vp158/files/CUB_200_2011.tgz?download=1"
236+
images_url="https://data.caltech.edu/records/65de6-vp158/files/CUB_200_2011.tgz?download=1",
237+
segmentations_url="https://data.caltech.edu/records/w9d68-gec53/files/segmentations.tgz?download=1"
237238
# images_url="https://drive.google.com/uc?export=download&id=1hbzc_P1FuxMkcabkgn9ZKinBwW683j45",
238239
# split_url=None,
239240
# annotations_url="https://drive.google.com/uc?export=download&id=1EamOKGLoTuZdtcVYbHMWNpkn3iAVj8TP",
@@ -374,12 +375,13 @@ class CaltechBirdsInfo(
374375
class CaltechBirdsInfo2011(
375376
collections.namedtuple(
376377
"_CaltechBirdsInfo2011",
377-
["name", "combined_url"],
378+
["name", "images_urls", "segmentations_url"],
378379
)
379380
):
380381
"""Contains the information necessary to generate a Caltech Birds 2011 dataset.
381382
382383
Args:
383384
name (str): name of dataset.
384-
combined_url (str): URL containing images and attributes.
385+
images_urls (str): URL containing images
386+
segmentations_url (str): URL containing segmentations.
385387
"""

0 commit comments

Comments
 (0)