Pre-trained Knowledge Distillation Learning for Fine-grained Categorization
This project contains the below files:
data_utils.py --- dataset and dataloader for CUB-200 and Oxford Flower Dataset
main.py --- the training and test script
baseline.py --- baseline models
model.py --- the implementation of our own model
options.py --- Experiment settings
transformer_networks.py --- the components help to build our own model