Birdbrain

Citation

Birdbrain is trained on the Caltech-UCSD Birds-200-2011 dataset. If you use the dataset or this project in published work, please cite the sources below.

CUB-200-2011 dataset

The Caltech-UCSD Birds-200-2011 Dataset
Wah, C.; Branson, S.; Welinder, P.; Perona, P.; Belongie, S.
California Institute of Technology, 2011.
Technical Report CNS-TR-2011-001

Dataset home: https://www.vision.caltech.edu/datasets/cub_200_2011/

CUB images are restricted to non-commercial research and educational use. See the official dataset website for terms.

BibTeX (dataset)

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

Model weights

Checkpoints use ImageNet-pretrained backbones from torchvision (EfficientNet-B0, ResNet50). Cite torchvision / the original backbone papers if you publish results that depend on those weights.