AI in Ophthalmology

Diabetic Retinopathy Detection

import torchvision.models as models

class DiabeticRetinopathyClassifier(nn.Module):
    def __init__(self, n_grades=5):
        super().__init__()
        self.backbone = models.efficientnet_b0(pretrained=True)
        self.backbone.classifier = nn.Sequential(
            nn.Dropout(0.3), nn.Linear(1280, 256), nn.ReLU(),
            nn.Dropout(0.2), nn.Linear(256, n_grades))

    def forward(self, x):
        return self.backbone(x)

Glaucoma Screening

Cup-to-Disc Ratio

Optic Disc Segmentation

class OpticDiscSegmenter(nn.Module):
    def __init__(self):
        super().__init__()
        self.encoder = nn.Sequential(
            nn.Conv2d(3, 64, 3, padding=1), nn.BatchNorm2d(64), nn.ReLU(),
            nn.MaxPool2d(2), nn.Conv2d(64, 128, 3, padding=1), nn.ReLU())
        self.decoder = nn.Sequential(
            nn.ConvTranspose2d(128, 64, 2, stride=2), nn.ReLU(),
            nn.Conv2d(64, 2, 1))

    def forward(self, x):
        return self.decoder(self.encoder(x))

OCT Analysis

Retinal Layer Segmentation

Evaluation