Learn practical skills, build real-world projects, and advance your career

COVID-19, Pneumonia and normal lung X-Ray Classification

###This notebook contains model to classify X-ray lung images of Covid-19, Viral Pneumonia patients and normal person using Resnets in pytorch.
Covid19-image-dataset is an image dataset in Kaggle used for classification into Covid-19, Viral Pneumonia and normal X-ray images.

!pip install jovian --upgrade --quiet
#Import required libraries
import os
import torch
import torchvision
import tarfile
import torch.nn as nn
import numpy as np
import torch.nn.functional as F
from torchvision.datasets.utils import download_url
from torchvision.datasets import ImageFolder
from torch.utils.data import DataLoader
import torchvision.transforms as tt
from torch.utils.data import random_split
from torchvision.utils import make_grid
import matplotlib
import matplotlib.pyplot as plt
%matplotlib inline

matplotlib.rcParams['figure.facecolor'] = '#ffffff'

Download kaggle dataset