Retinal disease classification
Classifies retinal fundus images with three model families — a ResNet-style CNN, a vision GNN that builds dynamic k-NN graphs over image patches, and a hybrid that feeds CNN features into graph layers. Predictions are served through a FastAPI inference API and a plain HTML/CSS/JS frontend.
- Binary disease-risk and multilabel detection of DR, MH, TSLN and ODC
- Trained and evaluated on the RFMiD dataset with selectable GCN, GAT and GIN graph layers
- Web demo with test-set gallery, image upload and adjustable thresholds
- Python
- PyTorch
- PyTorch Geometric
- FastAPI