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

Pet Image Classification using Convolutional Neural Networks in PyTorch

O Objetivo desse notebook é estudar o conteúdo do Video 4 do curso Zero to Gans da joviam.ml , o dataset a ser utilizado foi disponibilizado pela fastai chamado oxford-iiit-pet.

Neste Notebook criarei um modelo de classificação de imagem tentando classificar 37 categorias de pets com 200 imagens de cada classe .

Pacotes

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

Preparação dos dados