Sadou/malaria-detector-dinov2-tanzania

Malaria Detector — DINOv2 fine-tuned on Tanzania Blood Smears 🦟

Fine-tuned facebook/dinov2-base for binary classification of blood smear images: malaria infected vs healthy.

Trained on real patient data from 5 Tanzanian health centres, with rigorous handling of class imbalance.

📊 Performance (test set)

⚖️ Class imbalance handling

Dataset has 1.64x imbalance (Paludisme: 62% vs Sain: 38%).

Techniques applied :

1. 1.✅ Class weights in loss function
2. 2.✅ Stratified split on subtype (not just binary label)
3. 3.✅ Aggressive data augmentation
4. 4.✅ MCC-based model selection (instead of accuracy)
5. 5.✅ Multiple imbalance-robust metrics reported

🚀 Quick Start

from transformers import AutoImageProcessor, AutoModelForImageClassification
from PIL import Image
import torch

processor = AutoImageProcessor.from_pretrained('Sadou/malaria-detector-dinov2-tanzania')
model = AutoModelForImageClassification.from_pretrained('Sadou/malaria-detector-dinov2-tanzania')

img = Image.open('blood_smear.jpg').convert('RGB')
inputs = processor(img, return_tensors='pt')

with torch.no_grad():
    outputs = model(**inputs)

probs = torch.softmax(outputs.logits, dim=-1)[0]
pred = torch.argmax(probs).item()

labels = ['Healthy', 'Malaria']
print(f'{labels[pred]} (confidence: {probs[pred]:.1%})')

📂 Dataset

Tanzania Malaria Blood Smear Dataset

• —Source: Harvard Dataverse
• —DOI: 10.7910/DVN/O2WVWA
• —Images: 3,544 real blood smears
• —Source: 5 health centres in Tanga region, Tanzania
• —Microscope: 4K SONY IMX334 sensor (40X-2500X)
• —Staining: Giemsa reagent
• —Reference: Lufyagila et al. 2024, Data in Brief

Class distribution

⚠️ Disclaimer

This is a research tool, NOT a certified medical device. Any real diagnosis must be confirmed by a qualified microscopist or physician. Clinical validation is required before any real-world deployment.

📚 Citation

@misc{malaria-dinov2-2026,
  author = {Sadou Barry},
  title = {Malaria Detector — DINOv2 fine-tuned on Tanzania Blood Smears with Class Imbalance Handling},
  year = {2026},
  publisher = {HuggingFace},
  url = {https://huggingface.co/Sadou/malaria-detector-dinov2-tanzania}
}