Dementia Detection Model

This model detects dementia from brain MRI scans.

Model Details

• Architecture: EFFICIENTNET
• Task: Image Classification
• Classes: 4
• Input Size: 128x128

Labels

• 0: Non-Demented
• 1: Very-Mild-Demented
• 2: Mild-Demented
• 3: Moderate-Demented

Usage

from transformers import AutoFeatureExtractor, AutoModelForImageClassification
from PIL import Image

model_name = "dhineshkmar/dementia"
feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
model = AutoModelForImageClassification.from_pretrained(model_name)

image = Image.open("brain_scan.jpg")
inputs = feature_extractor(images=image, return_tensors="pt")
outputs = model(**inputs)
predicted_class = outputs.logits.argmax(-1).item()

Training

Trained on brain MRI scan dataset for dementia detection.