In this very short PyTorch project, we will build a Bilinear CNN Models For Fine-Grained Visual Recognition on FGVC-Aircraft dataset.
- Python3
- Other dependencies are in
requirements.txt
Download the FGVC-Aircraft dataset from url, then put and unzip fgvc-aircraft-2013b.zip
file into the data folder. Run the command below to generate a dataset for the training process.
python gen_data.py
We proposed 5 methods for the classifier training as follows,
- python main.py --type_cls softmax
- python main.py --type_cls cxsvm1
- python main.py --type_cls cxsvm2
- python main.py --type_cls cxsvm3
- python main.py --type_cls cxsvm4
tuantran23012000
Please create an issue or contact me through [email protected], thanks!
TuanTA