Learn practical skills, build real-world projects, and advance your career
project_name= "course-project"

Find a dataset

In this case we will be using the simpsons-faces dataset to train a GAN to generate simpson-like faces.

!pip install opendatasets --upgrade --quiet
import opendatasets as od
dataset_url = 'https://www.kaggle.com/kostastokis/simpsons-faces'
od.download(dataset_url)
Please provide your Kaggle credentials to download this dataset. Learn more: http://bit.ly/kaggle-creds Your Kaggle username: alexc957 Your Kaggle Key: ··········
0%| | 0.00/442M [00:00<?, ?B/s]
Downloading simpsons-faces.zip to ./simpsons-faces
100%|██████████| 442M/442M [00:05<00:00, 89.3MB/s]
!rm -rf ./simpsons-faces/simplified # we delete the simplified version of the faces 
from torch.utils.data import DataLoader
from torchvision.datasets import ImageFolder
import torchvision.transforms as T

DATA_DIR = './simpsons-faces'
image_size = 64
batch_size = 128
stats = (0.5, 0.5, 0.5), (0.5, 0.5, 0.5)

train_ds = ImageFolder(DATA_DIR, transform=T.Compose([
    T.Resize(image_size),
    T.CenterCrop(image_size),
    T.ToTensor(),
    T.Normalize(*stats)]))

train_dl = DataLoader(train_ds, batch_size, shuffle=True, num_workers=3, pin_memory=True)