Learn practical skills, build real-world projects, and advance your career
Updated 3 years ago
!pip install jovian --upgrade --quietWARNING: You are using pip version 20.3.1; however, version 20.3.3 is available.
You should consider upgrading via the '/opt/conda/bin/python3.7 -m pip install --upgrade pip' command.
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.pyplot as plt
%matplotlib inlineproject_name='course-project'data_dir = '../input'
print(os.listdir(data_dir))
classes = os.listdir(data_dir + "/sarscov2-ctscan-dataset")
print(classes) ['sarscov2-ctscan-dataset']
['non-COVID', 'COVID']
from torchvision.datasets import ImageFolder
from torchvision import transforms
from torchvision.transforms import ToTensor