Mohan Kumar mk-devc
mk-devc
/ customCelebA.py
Last active
March 5, 2021 08:08
This is a custom dataset for loading celebA dataset to be trained on google colab. Due to google drive limitations this could serve as a manual alternative.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Download the data on your google drive by making a copy of the annotations | |
| # change the shareable link file/d/ into uc?id=/ | |
| # store those links in their respective url | |
| # use the gdown command on the url to download straight to colab | |
| import pandas as pd | |
| import os | |
| from skimage import io | |
| from torch.utils.data import DataLoader |