GitHub - SahilC/knowledge-distill-fine-grained: Code for the Knowledge distillation work to enhance fine grained disease recognition.

Knowledge Distillation for Disease Prediction from Fundus Images

Work done in Microsoft AI Research.

Run the code with:-

python main.py

Run configuration can be modified in:-

config_small.gin

For Grad-cam visualizations, load the model you'd like to visualize on line 228 in grad-cam.py and run with:-

python grad-cam.py --use-cuda

The grad visualizations will be written to a folder outputs.

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
README.md		README.md
config.gin		config.gin
config_small.gin		config_small.gin
dataset.py		dataset.py
grad-cam.py		grad-cam.py
main.py		main.py
main_autoencoder.py		main_autoencoder.py
models.py		models.py
plot_tsne.py		plot_tsne.py
trainer.py		trainer.py
trainer_small.py		trainer_small.py
utils.py		utils.py