Ghastly faces and fake flowers
Assignment for TDT4265 - Computer Vision and Deep Learning
Make sure you have pytorch, torchvision and matplotlib.
Open gan.ipynb and run the whole notebook.
By default, the notebook will train on the flowers 102 dataset. To train on the faces, you need to download the dataset:
- Download the dataset zip-file
- Unzip it such that each frames path is
assets/frames/frameXXXX.png - Set
SELECTED_DATASETtovetleDatasetin the "Preparing training data"-section - Run the notebook