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

23 Pet Breed Classification from Scratch with PyTorch

Importing the neccesary Libraries

!pip install opendatasets --upgrade --quiet

import os
import torch
import tarfile
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
import opendatasets as od
import torchvision
import torch.nn as nn
import torch.nn.functional as F
import torchvision.transforms as T
import torchvision.transforms as tt

from PIL import Image
from tqdm.notebook import tqdm
from torchinfo import summary
from IPython.display import Image
from torchvision.datasets.utils import download_url
from torchvision.datasets import ImageFolder
from torchvision.transforms import ToTensor
from torch.utils.data import DataLoader
from torch.utils.data import random_split
from torchvision.utils import make_grid

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

Downloading and Checking the Dataset

dataset_url = 'https://www.kaggle.com/aseemdandgaval/23-pet-breeds-image-classification'
od.download(dataset_url)
Skipping, found downloaded files in ".\23-pet-breeds-image-classification" (use force=True to force download)