| library_name: keras | |
| tags: | |
| - image-classification | |
| - medical | |
| - oct | |
| - ophthalmology | |
| # OCT Image Classifier | |
| A deep learning model for classifying OCT (Optical Coherence Tomography) images into four categories: | |
| - CNV (Choroidal Neovascularization) | |
| - DRUSEN | |
| - NORMAL | |
| - DME (Diabetic Macular Edema) | |
| ## Model Details | |
| - **Framework:** Keras/TensorFlow | |
| - **Input:** 224x224 RGB images (adjust if different) | |
| - **Output:** 4-class probabilities | |
| - **Classes:** ['CNV', 'DRUSEN', 'NORMAL', 'DME'] | |
| ## Usage | |
| ```python | |
| from transformers import pipeline | |
| classifier = pipeline("image-classification", "sumitt8085/oct_classifier") | |
| result = classifier("oct_image.jpg") |