Giter Club home page Giter Club logo

dcgan-celeba's Introduction

Udacity project Face Generator using CelebA

This project was part of the Udacity Deep Learning Nano Degree Program that was teaching basics on building Generative Adversarial Networks (GANs).

In this project, we did build and train a custom GAN architecture on the CelebA dataset, leveraging the different skills learned during the course mentioned above.

Model Architecture

Deep Convolutional GANs

In this notebook, I've chosen to apply the DCGAN architecture to build a GAN using convolutional layers in the generator and discriminator. This is called a Deep Convolutional GAN, or DCGAN for short.

The DCGAN architecture was first explored in 2016 and has seen impressive results in generating new images. You can read the original paper, here.

Next to using Convolutional layers, I used label smoothing technique for the discriminator and the 'Two Times Update Rule' for the genrator part of the model.

Data Set

You’ll be training DCGAN on the Large-scale CelebFaces Attributes (CelebA) Dataset dataset.

These are color images of celebrities. In this course, Udacity provides a smaller subset of this data set with in total 32,600 pre-processed images. More details are provided by Udacity in the Jupyter notebook face_generator.ipynb.

Getting Started

So, our goal is to create a DCGAN that can generate new, realistic-looking images. We’ll go through the following steps:

  1. Extract the prepared subset of the celeb data set
  2. Define discriminator and generator networks
  3. Train the DCGAN network
  4. Visualize the loss over time and some sample, generated images

Requirements

It is recommended to train the model on GPU.

dcgan-celeba's People

Contributors

aockel avatar dependabot[bot] avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.