Downloads · 30 days
0
lukmanaj/brain-tumor-multimodal
brain-tumor-multimodal is a machine learning model from lukmanaj. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This model uses a multimodal DenseNet architecture to classify brain tumors based on CT and MRI scans. It was trained on paired data with class labels: Healthy or Tumour.
Downloads · 30 days
0
Access
Public
Updated May 24, 2025
Repo size
155 MB
Likes
0
Public
Click a slice to open those files.
.pth155 MB · 100%
From the Hugging Face model README
This model uses a multimodal DenseNet architecture to classify brain tumors based on CT and MRI scans. It was trained on paired data with class labels: Healthy or Tumour.
Healthy, Tumour)multimodal_brain_tumor_model.pth: Pretrained weightsfrom huggingface_hub import hf_hub_download
import torch
path = hf_hub_download("lukmanaj/brain-tumor-multimodal", "multimodal_brain_tumor_model.pth")
model.load_state_dict(torch.load(path))
| Epoch | Train Loss | Accuracy |
|---|---|---|
| 1 | 0.1552 | 94.82% |
| 5 | 0.0368 | 98.78% |
Note: The model shows signs of overfitting; further validation and augmentation are advised.
Aliyu, L. (2025). Brain Tumor Classification using Multimodal Deep Learning