CoolFace
Modelpublic

DrakeBlaze/RetinaVision-MNet

sourceHugging Facemitupdated 10mo agoView on Hugging Face
0likes
Model Card

🧠 RetinaVision-MNet

RetinaVision-MNet is a custom-trained MobileNetV2-based deep learning model for multi-class retinal disease detection. It predicts 10 retinal conditions from fundus images and includes Grad-CAM heatmaps to provide interpretable visual explanations for every prediction.

The model is trained entirely from scratch and is hosted on Hugging Face due to GitHub’s file-size limitations.


πŸ”₯ Key Features

  • β€”10-class retinal disease classification
  • β€”MobileNetV2 backbone β€” lightweight and efficient for medical imaging
  • β€”Grad-CAM interpretability for understanding model decisions
  • β€”Custom-trained model (.h5) using Keras / TensorFlow
  • β€”Optimized for FastAPI deployment with async inference
  • β€”Works seamlessly with secure JWT-protected backend

πŸ“¦ Usage

Download the model file from the Files and Versions tab and place it in your project:

python
from tensorflow.keras.models import load_model

model = load_model("mobile_model.h5")