Skip to content

MichaelSykesUK/face-generation-with-deep-convolutional-generative-adversarial-network

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Face Generation with Deep Convolutional Generative Adversarial Network

Summary

Face generation with deep convolutional generative adversarial network using PyTorch and Jupyter Notebook.

This notebook was originally submitted as a Udacity project in 2024.

License

The contents of this repository are covered under the MIT License.

Project Files

The project files are located in the Project Workspace and include the following files:

  • dlnd_face_generation_starter.ipynb
  • README.md
  • requirements.txt
  • tests.py
  • processed-celeba-small.zip

We highly recommend using the Project Workspace to complete your project; however, if you choose to not use the workspace, you can download the project files from the Project Workspace.

Instructions

Open the notebook file, dlnd_face_generation_starter.ipynb and follow the instructions. This project is organized as follows:

  • Data Pipeline: implement a data augmentation function and a custom dataset class to load the images and transform them.
  • Model Implementation: build a custom generator and a custom discriminator to make your GAN
  • Loss Functions and Gradient Penalty: decide on loss functions and whether you want to use gradient penalty or not.
  • Training Loop: implement the training loop and decide on which strategy to use

Each section requires you to make design decisions based on the experience you have gathered in this course. Do not hesitate to come back to a section to improve your model or your data pipeline based on the results that you are getting.

Building a deep learning model is an iterative process, and it's especially true for GANs! Good luck!

Submitting Your Project

For this project you will need to submit one file – dlnd_face_generation.ipynb

The full project may be submitted in two ways:

Project completed in Project Workspace:

  • Your project may be submitted directly via the Project Workspace by pressing the Submit button in the bottom right corner of the workspace.

Project completed outside of Project Workspace:

  • Your project may be submitted using the Project Submission page by pressing the Submit Project button in the top right corner of the page and following those directions.
  • You will need to create a zip file of the required project file and submit the zip file.