metadata

language: en
tags:
  - image-classification
  - brain-tumor-detection
  - medical-imaging
  - deep-learning
  - keras

Brain Tumor Detection Model

This model predicts the presence of brain tumors based on medical imaging data (e.g., MRI scans). It is designed to assist healthcare professionals in identifying tumors in brain scans, enhancing diagnostic capabilities.

Model Description

Architecture: Convolutional Neural Network (CNN) built using TensorFlow/Keras.
Input Shape: The model expects input images of shape (224, 224, 3), which corresponds to resized RGB images.
Output: The model outputs a probability distribution over two classes: malignant and non-malignant.

Training Data

The model was trained on a dataset of brain MRI images. The dataset includes both images with and without tumors.
Data Augmentation techniques were applied during training to improve model generalization.

Usage

Installation

To use this model, ensure you have the required libraries installed:

pip install tensorflow fastapi numpy pillow