Learn practical skills, build real-world projects, and advance your career
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, '.')
0it [00:00, ?it/s]
Downloading http://files.fast.ai/data/cifar10.tgz to ./cifar10.tgz
168173568it [00:01, 88002056.66it/s]
# Extract from archive
with tarfile.open('./cifar10.tgz', 'r:gz') as tar:
    tar.extractall(path='./data')
data_dir = './data/cifar10'

print(os.listdir(data_dir))
classes = os.listdir(data_dir + "/train")
print(classes)
['labels.txt', 'train', 'test'] ['ship', 'bird', 'cat', 'horse', 'deer', 'automobile', 'dog', 'airplane', 'frog', 'truck']
PATH = "data/cifar10/"
trn_dir, val_dir = PATH + 'train', PATH + 'test'