Installation
Make sure you have Python>=3.6 installed on your machine.
Install python dependencies by running:
pip install -r requirements.txt
Datasets:
Place the public data sets correspondingly in the following directories：
./datasets/CUB-200-2011
./datasets/FGVC_Aircraft1
./datasets/Stanford_cars
Train:
For CUB-200-2011 dataset:
python train_CUB_resnet50.py
or
python train_CUB_mobilenetv3
or
python train_CUB_densenet161
For FGVC_Aircraft dataset:
python train_Air_resnet50.py
or
python train_Air_mobilenetv3
or
python train_Air_densenet161
For Stanford_cars dataset:
python train_Car_resnet50.py
or
python train_Car_mobilenetv3
or
python train_Car_densenet161