/Xray-Covid-Detection

This is a flask application, used to detect Covid using xray of patients.

Primary LanguagePythonMIT LicenseMIT

Xray-Covid-Detection

Aim of this project is to detect Covid in patients using their X-Rays.
Classification Model: Resnet18
Deep Learning Framework: Pytorch
API Framework: Flask


Screenshots

App Screenshot


Features

  • Responsive Web App
  • Supports Every Operating System
  • Can be easily deployed
  • Easy to use architecture

Tech Stack

Client: HTML, CSS and Bootstrap

Server: Python


API Reference

Train the Model

  POST /train (type=json)
Parameter Type Description
batch_size int batch size
epochs int Number of epochs to train
num_workers int Number of workers
learning_rate int learning rate
momentum int momentum
dataset_path path/string Path where the data is stored
model_save_folder path/string Folder/Directory name to store model

Prediction

  POST /predict
Parameter Type Description
image_path path/string Required. Path of the image

Installation

Install dependencies for the project using pip

    cd <project-folder>
    pip install requirements.txt

Deployment

Project can be deployled on Heroku

  git add .
  git commit -m <message>
  git push heroku main

Refer to the official documentation for more details: https://devcenter.heroku.com/articles/getting-started-with-python


License

MIT

Author

@Rishabh1501

Support

For support, email at rishabhkalra1501@gmail.com .