This web application is built using Django and utilizes pre-trained machine learning models to classify medical images for brain tumor, chest X-ray, and melanoma cancer. Users can upload their medical images and get predictions on whether the images indicate the presence of cancer.
- Installation
- Usage
- Features
- Technologies Used
- Model Information
- Data Sources
- Docker pull
- Python 3.11.2
- Django 4.2.3
- TensorFlow (for using the machine learning models)
- Other Python libraries as specified in the requirements.txt
-
Clone this repository to your local machine.
git clone https://github.com/harshkasat/Diagnose.git -
Create a virtual environment:
python -m venv .venv -
Activate virtual environment:
.\.venv\Scripts\activate -
Install the required packages:
pip install -r requirements.txt -
Change dir for Django development server:
cd diagnose -
Run the Django development server:
python manage.py runserver -
Access the application at http://localhost:8000/
- Upload a medical image for classification.
- The application will use the pre-trained machine learning models to make predictions.
- The results will be displayed, indicating whether the image shows signs of brain tumor, chest X-ray abnormalities, or melanoma cancer.
- Image upload and classification for brain tumor detection.
- Image upload and classification for chest X-ray abnormalities.
- Image upload and classification for melanoma cancer detection.
- Django: Web framework for backend development.
- HTML, CSS: Frontend development.
- TensorFlow: Machine learning library for the models.
- Python: Backend and model implementation.
- Brain Tumor Model: Trained on a dataset of brain MRI images using CNN architecture.
- Chest X-ray Model: Trained on a dataset of chest X-ray images using transfer learning with a pre-trained model.
- Melanoma Cancer Model: Trained on a dataset of skin lesion images using CNN architecture.
- Brain Tumor Dataset: Brain Tumor
- Chest X-ray Dataset: Pneumonia
- Melanoma Cancer Dataset: Melanoma Cancer
- Install Docker
docker pull zedmate/python-django