An-Erudite-FGVC-Model

Code release for “An Erudite Fine-Grained Visual Classification Model" (CVPR 2023)

Changelog

• 2023/04/18 upload the code.

Requirements

• python 3.6
• PyTorch 1.7.1+cu110
• torchvision 0.8.2+cu110
• learn2learn 0.1.7

Data

• Download datasets
• Extract them to data/cars/, data/birds/ and data/airs/, respectively.
• Split the dataset into train and test folder, the index of each class should follow the Birds.xls, Air.xls, and Cars.xls

• e.g., CUB-200-2011 dataset

  -/birds/train
	         └─── 001.Black_footed_Albatross
	                   └─── Black_Footed_Albatross_0001_796111.jpg
	                   └─── ...
	         └─── 002.Laysan_Albatross
	         └─── 003.Sooty_Albatross
	         └─── ...
   -/birds/test	
             └─── ...         

Training

• python main.py (Mix Cars and Flowers)

Citation

If you find this paper useful in your research, please consider citing:

Contact

Thanks for your attention! If you have any suggestion or question, you can leave a message here or contact us directly:

• changdongliang@bupt.edu.cn
• mazhanyu@bupt.edu.cn