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

Image classification using CNN

Credits: This notebook is based on https://jovian.ml/aakashns/cnn-workshop-live

import os 
import torch 
import torchvision 
import tarfile
from torchvision.datasets.utils import download_url
# Dowload the dataset
dataset_url = "http://files.fast.ai/data/cifar10.tgz"
download_url(dataset_url, './')
Using downloaded and verified file: ./cifar10.tgz
with tarfile.open('./cifar10.tgz','r:gz') as tar:
    tar.extractall(path='./data')