from shifaa.vision import VisionModelFactory

model = VisionModelFactory.create_model(
    model_type="classification",
    model_name="Eye_Disease"
)

result = model.run("eye_image.jpg", show_image=True)
print(f"Disease: {result['predicted_class']}")
print(f"Confidence: {result['confidence']:.2f}%")

Confusion Matrix:

Preprocessing:

Resize to 224×224
Convert to tensor
ImageNet normalization

Training Details:

Loss Function: CrossEntropyLoss
Optimizer: Adam (lr=0.0001)
Scheduler: ReduceLROnPlateau (factor=0.5, patience=3)

Ahmed-Selem/Shifaa-Eye-Disease-EfficientNetB0

Eye Disease Model

Ahmed-Selem/
Shifaa-Eye-Disease-EfficientNetB0