oct_classifier / README.md
sumitt8085's picture
Rename readme.md to README.md
6d64bff verified
metadata
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

from transformers import pipeline

classifier = pipeline("image-classification", "sumitt8085/oct_classifier")
result = classifier("oct_image.jpg")