/knowledge-distill-fine-grained

Code for the Knowledge distillation work to enhance fine grained disease recognition.

Primary LanguagePython

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.