File size: 1,003 Bytes
d9b0e69 f6368f7 d9b0e69 f6368f7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
---
license: mit
tags:
- medical
- brain-disease
- image-classification
- multiplesclerosis
---
# Multiplesclerosis Detection Model
This model detects multiplesclerosis from brain MRI scans.
## Model Details
- **Architecture**: VIT
- **Task**: Image Classification
- **Classes**: 4
- **Input Size**: 224x224
## Labels
- 0: Control-Axial
- 1: Control-Sagittal
- 2: MS-Axial
- 3: MS-Sagittal
## Usage
```python
from transformers import AutoFeatureExtractor, AutoModelForImageClassification
from PIL import Image
model_name = "dhineshkmar/multiplesclerosis"
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 multiplesclerosis detection.
|