This project develops a deep learning model using ResNet architecture to detect malaria-infected cells from microscopic images. The model classifies cell images as either "Parasitized" or "Uninfected" with high accuracy.
- Custom ResNet architecture for medical image classification
- Data augmentation to improve model generalization
- Streamlit web application for easy model deployment
- Comprehensive training and evaluation pipeline
- Source: Cell Images for Detecting Malaria
- Contains microscopic images of blood cells
- Binary classification: Parasitized vs. Uninfected
- Residual Network (ResNet) with custom stages
- Key components:
- Residual modules
- Batch normalization
- Dropout for regularization
- Softmax classification layer
- Python
- TensorFlow/Keras
- scikit-learn
- OpenCV
- Streamlit
- Python 3.8+
- pip
- Clone the repository
- Install dependencies:
pip install -r requirements.txt
- Download the dataset
- Run training script:
python train.py
- Launch Streamlit app:
streamlit run app.py
- Accuracy: 96%
- Precision, Recall: 97%, 98%
- Local deployment via Streamlit
- Potential cloud deployment on platforms like Heroku or AWS
- Expand dataset
- Experiment with more advanced architectures
- Add multi-class detection capabilities
[MIT]