1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17

import torch
from torchvision import datasets, transforms

# Root directory for the dataset
data_root = 'data'
# Spatial size of training images, images are resized to this size.
image_size = 64

celeba_data = datasets.CelebA(
        data_root,
        download=True,
        transform=transforms.Compose([
        transforms.Resize(image_size),
        transforms.CenterCrop(image_size),
        transforms.ToTensor(),
        transforms.Normalize(mean=[0.5, 0.5, 0.5],
                             std=[0.5, 0.5, 0.5]) ] ) )

1
2
3
4
5
6
7
8
9

import zipfile

data_root = 'data/celeba'

# Add shortcut of dataset to your google drive
zip_path = '/content/drive/MyDrive/CelebA/Img/img_align_celeba.zip'

with zipfile.ZipFile(zip_path, 'r') as ziphandler:
    ziphandler.extractall('data')

1
2
3
4

import gdown
url = "https://drive.google.com/u/0/uc?id=1m8-EBPgi5MRubrm6iQjafK2QMHDBMSfJ&export=download"
output = "celeba.zip"
gdown.download(url, output)

1
2
3

unzip celeba.zip
cd celeba
unzip img_align_celeba.zip

1
2
3
4
5
6
7
8
9

(base) zi@lambda-server:~/Downloads$ gdown 135oKPefcPTsdtLRzoDAQtPpHuoIrpRI_
Access denied with the following error:

        Cannot retrieve the public link of the file. You may need to change
        the permission to 'Anyone with the link', or have had many accesses. 

You may still be able to access the file from the browser:

         https://drive.google.com/uc?id=135oKPefcPTsdtLRzoDAQtPpHuoIrpRI_ 

1

pip install --upgrade gdown