My Car Classification Model - Notebook by Abubakar A Sudi (abubakarsoud97)

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

Updated 3 years ago

!pip install jovian --upgrade --quiet

import os
import torch
import torchvision
import tarfile
from torchvision.datasets.utils import download_url
from torch.utils.data import random_split

project_name='my-car-classification-model'

# Dowload the dataset
dataset_url = "https://s3.amazonaws.com/fast-ai-imageclas/stanford-cars.tgz"
download_url(dataset_url, '.')

Downloading https://s3.amazonaws.com/fast-ai-imageclas/stanford-cars.tgz to ./stanford-cars.tgz

HBox(children=(FloatProgress(value=1.0, bar_style='info', max=1.0), HTML(value='')))

from torchvision.datasets import ImageFolder
from torchvision.transforms import ToTensor
from torch.utils.data import Dataset