Skip to content

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.

License

Notifications You must be signed in to change notification settings

vantrn/Pneumonia-Detection

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

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages