Downloads · 30 days
24
11% of all-time downloads
diabetic-retinopathy-detection
ArjTheHacker/diabetic-retinopathy-detection
diabetic-retinopathy-detection is a machine learning model from ArjTheHacker. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This model is designed to detect and classify diabetic retinopathy from retinal images. It provides both color and black & white image analysis capabilities.
All-time downloads
216
Public
Repo size
43.1 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin43.1 MB · 100%
At a glance
- Model type
- vision-classification
- Access
- Public
- Created
- Jan 12, 2025
- Updated
- Jan 12, 2025
- SHA
- 8b214145
Spaces
Model card
From the Hugging Face model README
Diabetic Retinopathy Detection Model
This model is designed to detect and classify diabetic retinopathy from retinal images. It provides both color and black & white image analysis capabilities.
Model Description
The model comes in two variants:
- Color image model (
the_full_color_model.pth) - Black & White image model (
the_full_BW_model.pth)
Input
- Image size: [Please specify the input image size requirements]
- Format: Both RGB and grayscale images supported
Output
- Classification of diabetic retinopathy severity
- Confidence scores for each class
Usage
from transformers import AutoModelForImageClassification
model = AutoModelForImageClassification.from_pretrained("ArjTheHacker/diabetic-retinopathy-detection")
Training
The model was originally trained on [specify dataset] and fine-tuned for diabetic retinopathy detection.
Performance
[Add performance metrics when available]
Limitations
This model is intended to assist in diabetic retinopathy screening but should not be used as the sole diagnostic tool. Always consult healthcare professionals for medical decisions.
- Type
- vision-classification
- Created
- Jan 12, 2025
- Updated
- Jan 12, 2025