Learn practical skills, build real-world projects, and advance your career
Created 3 years ago
#Downloading Dataset
Opendatasets library is insatlled and it is used to download the Brain Tumor Classification (MRI) datset from Kaggle.
Link to the dataset
!pip install opendatasets --upgrade --quiet
import opendatasets as od
dataset_url = 'https://www.kaggle.com/sartajbhuvaji/brain-tumor-classification-mri/download'
od.download(dataset_url)
6%|▌ | 5.00M/86.8M [00:00<00:01, 45.9MB/s]
Downloading brain-tumor-classification-mri.zip to ./brain-tumor-classification-mri
100%|██████████| 86.8M/86.8M [00:01<00:00, 83.7MB/s]
Importing the necessary 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
from torchvision.transforms import ToTensor
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'