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

Planet Image Classification using Convolutional Neural Networks in PyTorch

import os
import torch
import torchvision
import tarfile
from torchvision.datasets.utils import download_url
from torch.utils.data import random_split
import pandas as pd
import matplotlib.pyplot as plt

project_name='assignment-5'

Exploring the Data

We'll download the images in PNG format from this page, using some helper functions from the torchvision and tarfile packages.

# Dowload the dataset
dataset_url = "http://files.fast.ai/data/examples/dogscats.tgz"
download_url(dataset_url, '.')
Using downloaded and verified file: .\dogscats.tgz