This repository provides an implementation of the most renowned FGVC (Fine-Grained Visual Categorization) datasets, complete with a training script using ResNet50. Some datasets, originally unavailable or with altered structures, have been reorganized and are now accessible via Google Drive for automated downloading.
The purposes of this project are
- providing unified interfaces to FGVC datasets
- easy access to dataset using auto downloading before training
- expendable to new datasets
The scripts have been tested on CUDA 11.8 with torch 2.0.
Datasets listed:
- FGVC Aircraft
- Caltech UCSD Birds (CUB-200-2011)
- Stanford Cars
- Stanford Dogs
- North American Birds (NABirds)
- iNaturalist 2017(Inat2017)
- TinyImageNet
This project is compatible with both Linux and Windows operating systems.
# Clone the repository:
git clone https://github.com/qiyuliaogh/FGVC_baselines_pytorch.git
cd FGVC_baselines_pytorch
# Install dependencies:
pip install torch numpy time tqdm albumentations torchvision json pandas cv2 random scipy matplotlibThen modify the including in training script, for example, if you want to use Stanford dogs datasset, use:
from datasets.Dogs import Dogs as FGVC_DatasetUncomment this line and comment out the rest.
python model_trainer.py --device cuda:0