Giter Club home page Giter Club logo

covid19-xray-classifier's Introduction

COVID19-xray-classifier

A classifier that takes X-RAY or CT-SCAN to classify positive or negative COVID19 with 99% accuracy, 97% recall (macro avg) and 91% precision (macro avg).

Data Preparation

The data for training the model was gathered from two sources:

  1. COVID19 X-RAY images repository: https://github.com/ieee8023/covid-chestxray-dataset
  2. Chest X-Ray Images (Pneumonia) on Kaggle: https://www.kaggle.com/paultimothymooney/chest-xray-pneumonia

The preprocessing and cleaning of images/labels was done to ensure data in correct form. For further reference of the process please check this notebook https://www.kaggle.com/rgaltro/covid19-data-preprocessing.

Training of model

Different models were trained like VGG16, Autoencoder classifier and InceptionV3. In the end InceptionV3 produced the best results.

Note:

To tackle the high imblance of data, different class weights were used.

Evalutation and results

The dataset was divided into train, validation and test set beforhand to ensure the sanity of further process.

The model was selected while training which had the highest val_accuracy.

Sample output of the test-set images:

COVID-19 Positive patient COVID-19 Negative patient COVID-19 Positive patient

Confusion matrix for test-set images

Confusion Matrix non-norm Confusion Matrix norm

Classification report of the trained model on test-set images

Classification Report

ROC curve

ROC curve

Conclusion

The InceptionV3 model with modified head converged to classifiy the prositive and the negative cases of COVID19 with 99% accuracy, 97% recall (macro avg) and 91% precision (macro avg).

Further work

The model can be further improved with more COVID19 positive cases for training.

Note:

This work was inspired from the work of Hanan Gani work at: https://github.com/hananshafi/covid19-detection .

covid19-xray-classifier's People

Contributors

rohilrg avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.