zenith0923 / C2-Net

5 stars 1 forks source link

Cross-Layer and Cross-Sample Feature Optimization Network for Few-Shot Fine-Grained Image Classification

Alt text

Data Preparation

The following datasets are used in our paper:

Stanford Dogs: Dataset Page

Stanford Cars: Dataset Page

CUB_200_2011: Dataset Page

iNaturalist2017 : Dataset Page

Please proceed with the setting up data by referring to FRN Github.

Usage

Requirement

All the requirements to run the code are in requirements.txt.

You can download requirements by running below script.

pip install -r requirements.txt

Train

Running the shell script run.sh will train and evaluate the model with hyperparameters matching our paper.