Downloads · 30 days
4
5% of all-time downloads
EnginCN55/brain_tumor_classification
brain_tumor_classification is a image classification model from EnginCN55. Use it when you need a label for an image. The card lists the license as mit.
Model Summary This model card describes two deep learning models trained to classify brain tumor MRI images into different tumor types. The models are based on ResNet50 and MobileNetV2 architectures and were trained u…
Downloads · 30 days
4
5% of all-time downloads
All-time downloads
79
Public
Repo size
26.4 MB
Likes
0
Public
Click a slice to open those files.
.tflite26.4 MB · 100%
From the Hugging Face model README
Model Summary This model card describes two deep learning models trained to classify brain tumor MRI images into different tumor types. The models are based on ResNet50 and MobileNetV2 architectures and were trained using the Brain Tumor MRI Dataset available on Kaggle. They aim to assist medical professionals in detecting brain tumors using transfer learning approaches. Model Details ResNet50
MobileNetV2
Uses Direct Use These models can be used to classify MRI scans for brain tumor detection in clinical decision-support systems. Out-of-Scope Use Not intended for standalone diagnostic purposes without medical supervision. Misuse includes deployment without validation or interpretability assessments. Bias, Risks, and Limitations The model performance may vary depending on the image quality, scanner differences, and patient demographics. Models may inherit biases from the training data. Training Details Training Data The models were trained on the Brain Tumor MRI Dataset from Kaggle. The dataset contains images categorized into three classes: glioma, meningioma, and pituitary tumors. Training Hyperparameters
Evaluation Testing Data Test data is a stratified split of the original dataset with unseen examples from each tumor class. Metrics Accuracy, Precision, Recall, F1-score Results
Environmental Impact
Technical Specifications Model Architecture and Objective Transfer learning using pretrained CNNs (ResNet50 and MobileNetV2) adapted for multi-class classification of brain tumor MRI images. Software Python 3.10, TensorFlow/Keras, NumPy, Matplotlib, OpenCV