This repo contains some unofficial PyTorch dataset APIs(mainly for Fine-Grained Visual Categorization task), which support automatically download (except large-scale datasets), extract the archives, and prepare the data.
- CUB-200-2011
- Stanford Dogs
- Stanford Cars
- FGVC Aircraft
- NABirds
- Tiny ImageNet
- iNaturalist 2017
- Oxford 102 Flowers
- Oxford-IIIT Pets
- Food-101
The code was tested on
- pytorch==1.4.0
- torchvision==0.4.1
Use them the similar way you use torchvision.datasets.
train_dataset = Cub2011('./cub2011', train=True, download=False)
test_dataset = Cub2011('./cub2011', train=False, download=False)Feel free to open an issue or PR.
MIT