Skip to content

Training a neural network model to detect covid-19 from chest X-rays

License

Notifications You must be signed in to change notification settings

masoudhassani/chest_xray_covid_19

Repository files navigation

Pneumonia detection from chest x-ray

Training a neural network model to detect Pneumonia from chest X-rays Note: This tool was solely developped for exploring the usability of deep learning in identifying Pneumonia from chest x-ray scans. This CANNOT be used for medical purposes in any way!

How to start

Download the chest x-ray dataset and put it in the project folder

https://www.kaggle.com/paultimothymooney/chest-xray-pneumonia/

for 2-label training (normal and pneumonia)

run train_normal_pneumonia.ipynb

for 3-label training (normal, viral and bacterial)

run train_normal_viral_bacterial.ipynb

to test your model

run test_image.py, then choose your saved model and an image

Requirements

opencv

pip install opencv-python

tensorflow

pip install --upgrade tensorflow

Misc

pip install pyh5
pip install matplotlib
pip install numpy
sudo apt-get install python3-tk

About

Training a neural network model to detect covid-19 from chest X-rays

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published