Skip to content

beingaryan/Chest-Xray-Pneumonia-Prediction

Repository files navigation

Chest-Xray-Pneumonia-Prediction

Chest-Xray-Pneumonia-Prediction built with OpenCV, Keras/TensorFlow using Deep Learning and Computer Vision concepts.

                               Python contributions welcome Forks Stargazers Issues LinkedIn

                                   

Non Pneumonia

👇 Support me here!

Buy Me A Coffee

😇 Motivation

A Deep Learning based model used for the prediction whether a person is suffering from pneumonia or not. The project is absed upon the Standard Convolutional Based Neural Network Architectural Implementation. It incorporates the use of CNN layers with Hyper-parameters tuning. The motivation behind the project is to effectively classify the reports of chest xrays to classify into pneumonia or non-pneumonia cases.

The model has been trained for 12 epoch. The model gave an accuracy of 98.51 on validation sets.

Case of Pneumonia

:⚠: TechStack/framework used

📁 Data Distribution

The dataset used can be downloaded here - Click to Download

This dataset consists of images belonging to 2 classes:

⭐ Features

Our model is capable of predicting Pneumonia from chest x-ray images with high efficiency. These predicted images are converted to grayscale version for predictions.

The model is efficient, since we used a compact CNN-based architecture, it’s also computationally efficient and thus making it easier to deploy the model to servers.

🔑 Prerequisites

All the dependencies and required libraries are included in the file requirements.txt See here

🚀  Installation

  1. Start and fork the repository.

  2. Clone the repo

$ git clone https://github.com/beingaryan/Chest-Xray-Pneumonia-Prediction.git
  1. Change your directory to the cloned repo and create a Python virtual environment named 'test'
$ mkvirtualenv test
  1. Now, run the following command in your Terminal/Command Prompt to install the libraries required
$ pip3 install -r requirements.txt

💡 Working

  1. Open terminal. Go into the cloned project directory and type the following command:
$ python3 jupyter
  1. To train the model, open the Pneumonia_Prediction file in jupyter notebook and run all the cells



🔑 Results

Our model gave 98.51% accuracy for validation set of Pneumonia Detection via tensorflow-gpu==2.0.0


* The model has been trained on a python based environment on Jupyter platform. * The model is iterated for a total epoch of 12. * The model has attained an accuracy of __98.51 %__ accuracy on the Validation set.

We got the following accuracy vs. epochs curve plot


The above figure shows the Accracy plot of the model throughout it's training journey.




The above figure shows the Loss plot of the model throughout it's training journey.

👏 And it's done!

Feel free to mail me for any doubts/query :email: aryan.gupta18@vit.edu

🤝 Contribution

Feel free to file a new issue with a respective title and description on the the Pneumonia_Detection repository. If you already found a solution to your problem, I would love to review your pull request!

❤️ Owner

Made with ❤️  by Aryan Gupta

👍 Credits

🤝 Our Contributors

CONTRIBUTORS.md

👀 Code of Conduct

You can find our Code of Conduct here.

👀 License

MIT © Aryan Gupta

About

Chest-Xray-Pneumonia-Prediction built with OpenCV, Keras/TensorFlow using Deep Learning and Computer Vision concepts.

Topics

Resources

License

Code of conduct

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published