bentrevett/caltech-ucsd-birds-200-2011

Caltech-UCSD Birds-200-2011 (CUB-200-2011)

[image]

This dataset contains the Caltech-UCSD Birds-200-2011 (CUB-200-2011) dataset, from here.

Each example consists of an image, a label, and a bounding box. (The dataset also contains x/y locations of "parts", e.g. beak, right eye, left wing, throat, etc. and "attributes", e.g. beak shape, wing color, feather pattern. I have not included either of these. Contact me if you want me to add them.)

Note: Some of these images are also in ImageNet!

Data Splits

The CUB-200-2011 dataset has 2 splits: train and test.

There are 200 classes, with either 29-30 examples per class in the train split. The test split has more variance in the number of examples per class; most are 29-30 but there are some with fewer (the lowest is 11).

Bounding Boxes

Each bounding box is in the form of [x0, y0, x1, y1] and can be used as such:

import datasets
from PIL import Image, ImageDraw
import matplotlib.pyplot as plt

dataset = datasets.load_dataset("bentrevett/caltech-ucsd-birds-200-2011")
example = dataset["train"][0]

image = example["image"]
bbox = example["bbox"]
draw = ImageDraw.Draw(image)
draw.rectangle(bbox, outline="red", width=2)
plt.imshow(image)

[image]

Citation Information

@techreport{WahCUB_200_2011,
	Title = The Caltech-UCSD Birds-200-2011 Dataset,
	Author = {Wah, C. and Branson, S. and Welinder, P. and Perona, P. and Belongie, S.},
	Year = {2011}
	Institution = {California Institute of Technology},
	Number = {CNS-TR-2011-001}
}