Skip to content

This project comprises predicting different types of disease at one place Pneumonia, Malaria, Liver Disease and Cardiovascular Disease

License

Notifications You must be signed in to change notification settings

Data-Science-Community-SRM/Ailment

Repository files navigation

Disease Predictor

Disease prediction for Pneumonia, Malaria, Liver Disease and Cardiovascular Disease

Flask Application Output

  • Sample output for Pneumonia Prediction using Lung X-ray images

Web app output #1

Web app output #2


  • Sample output for Malaria Prediction using cell images

Web app output #3

Web app output #4


The Flask Web Application

  • The Web Application has been built with Flask in the backend and HTML and Bootstrap for the frontend.
  • Respective images (.jpg, .jpeg, .png) can be uploaded to get predictions for Pneumonia and Malaria.
  • Functionality for preventing upload of file of any format other than .jpg, .jpeg and .png has been included.
  • Sample images are included in the Sample images folder.

Execution

  • The prerequisites for running the Flask Application are included in the requirements.txt file.
  • To run the application:
export FLASK_APP=run.py
export FLASK_ENV=development
flask run

The Models

Pneumonia Model

  • The Chest X-Ray Images (Pneumonia) dataset from Kaggle has been used to train this model.
  • This model has been trained to identify 2 classes (Positive or negative for Pneumonia) using transfer learning on the InceptionV3 model till layer ‘mixed7’, followed by a Dense layer with 512 nodes (RelU), and a sigmoid layer with 1 output node using Keras with Tensorflow backend.
  • It was trained using the RMSprop optimizer with a batch size of 64. Input size of the images were (150, 150, 3). The images were rescaled before training. (trainPneumonia.ipynb)
  • The final trained model resulted in an accuracy of 85.2% on the test set with 777 images.
  • Each image is resized to 150x150 and then normalized before feeding into the network to make a prediction.

Malaria Model


Contribute

Thanks for taking the time to contribute!

The following is a set of guidelines for contributing to Disease Predictor. Please check out the Contributing.md. These are just guidelines, not rules, so use your best judgement and feel free to propose changes to this document in a pull request. If you have any questions, open an issue.


License

MIT © Data Science Community SRM

This project is licensed under the MIT License - see the License.md file for details

License

Made with ❤️ by DS Community SRM

About

This project comprises predicting different types of disease at one place Pneumonia, Malaria, Liver Disease and Cardiovascular Disease

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published