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.