Home

Awesome

Vision Mamba Distillation for Low-resolution Fine-grained Image Classification

Configuration

Python 3 Pytorch 2.1

Training

./ViMD_train_teacher.sh

Please modify the model path in ViMD_train_teacher.sh and ViMD_train_teacher.py

./ViMD_train_student.sh

Please modify the model path in ViMD_train_student.sh and ViMD_train_student.py

Testing

./ViMD_test.sh