项目作者: mohd-faizy

项目描述 :
Detecting COVID-19 with Chest X-Ray using PyTorch from COVID-19 Radiography Dataset on Kaggle
高级语言: Jupyter Notebook
项目地址: git://github.com/mohd-faizy/09P_Detecting_COVID_19_with_Chest_X-Ray_using_PyTorch.git


author
made-with-Markdown
Language
Platform
Maintained
Last Commit
GitHub issues
Open Source Love svg2
Stars GitHub
GitHub license
Size

Detecting COVID 19 with Chest X-Ray using PyTorch



head_image

This repository is a case study for detecting COVID-19 with Chest X-Ray using PyTorch from COVID-19 Radiography Dataset on Kaggle

Dataset

Chest X-ray images for COVID-19 dataset:-

In our current release, there are 1143 COVID-19 positive images, 1341 normal images and 1345 viral pneumonia images.

  1. # Create api key -> account ->Create New Token
  2. # .json file is genrated
  3. !pip install -q kaggle
  4. from google.colab import files
  5. files.upload() #upload kaggle.json
  1. !mkdir -p ~/.kaggle
  2. !cp kaggle.json ~/.kaggle/
  3. !ls ~/.kaggle
  4. !chmod 600 /root/.kaggle/kaggle.json
  5. # API Command
  6. !kaggle datasets download -d tawsifurrahman/covid19-radiography-database

Objective

  • Create custom Dataset and DataLoader in PyTorch
  • Train a ResNet-18 model in PyTorch to perform Image Classification

Installation

Clone this repository:

  1. git clone https://github.com/mohd-faizy/09P_Detecting_COVID_19_with_Chest_X-Ray_using_PyTorch.git

or click Download ZIP in right panel of repository and extract it.

Open latest version of notebook in Jupyter Notebook.

Project Structure

  • Task :zero::one: Importing the Dataset from Kaggle
  • Task :zero::two: Importing Libraries
  • Task :zero::three: Preparing Training and Test Sets
  • Task :zero::four: Creating Custom Dataset
  • Task :zero::five: Image Transformations
  • Task :zero::six: Prepare DataLoader
  • Task :zero::seven: Data Visualization
  • Task :zero::eight: Creating the Model
  • Task :zero::nine: Training the Model
  • Task :one::zero: Show the Predictions
  • Task :one::one: Saving the Model
  • Task :one::two: Inference on a Single Image

Model Training Privew



head_image

Prediction Preview



head_image

Connect with me:

codeSTACKr | Twitter
codeSTACKr | LinkedIn
codeSTACKr.com


Faizy's github stats

Top Langs