Instructions to use linakosmina/densenet121-brain-tumor-classification with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use linakosmina/densenet121-brain-tumor-classification with Keras:
# !pip install -U keras tensorflow huggingface_hub # Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here; # "jax" and "torch" also work for computation once TensorFlow is installed. import os os.environ["KERAS_BACKEND"] = "tensorflow" import keras model = keras.saving.load_model("hf://linakosmina/densenet121-brain-tumor-classification") - Notebooks
- Google Colab
- Kaggle
Model Description.
This model is a fine-tuned DenseNet121 for classifying brain tumors from MRI scans. It was trained using transfer learning on a publicly available dataset from Kaggle. The model is capable of distinguishing between four classes: pituitary, glioma, meningioma, and no_tumor.
Intended Use.
This model is intended for non-commercial and research purposes only. It can be used for:
- Classifying brain tumors in MRI images.
- Serving as a baseline for further research.
Performance.
The model was evaluated on a held-out test set and achieved the following results:
Metric.
Accuracy 98.45% Precision 98% Recall 98% F1-Score 98%
Training Data.
The model was trained on the Brain Tumor MRI Dataset available on Kaggle.
Training Details.
- Base Model: DenseNet121 (pre-trained on ImageNet)
- Framework: TensorFlow / Keras
- Training Platform: Kaggle Notebooks
- Downloads last month
- 2