Learn practical skills, build real-world projects, and advance your career
Updated 3 years ago
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)