Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pneumonia Detection from Chest X-rays 🫁

This project uses a Convolutional Neural Network (CNN) to detect pneumonia from chest X-ray images. The model is trained on a labeled dataset and classifies images as either NORMAL or PNEUMONIA.

The main goals of this project are:

  • Build and train a CNN using TensorFlow/Keras
  • Evaluate predictions on test images
  • Visualize model output using matplotlib
  • Eventually implement Grad-CAM to understand which parts of the image the model focuses on

🚧 Status: Work In Progress

This project is still in progress. Currently, the model can make predictions, but:

  • Grad-CAM is not yet implemented
  • Model performance may be affected by dataset imbalance
  • Improvements and testing are ongoing

✅ How to Run

  1. Clone the repo and set up a Python virtual environment:

    python -m venv venv
    source venv/bin/activate  # or venv\Scripts\activate on Windows
    pip install -r requirements.txt
    
  2. Download and extract the Chest X-Ray Pneumonia dataset into the data/ folder Text to display

  3. Train the model

python main.py
  1. Predict with test images
python evaluate.py

About

Uses CNN to detect pneumonia from chest X-ray images. The model is trained on a labeled medical dataset and classifies images as either normal or pneumonia. It includes data preprocessing, model training, evaluation, and image prediction, and is built using TensorFlow and Python.

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages