Home

Awesome

DCGAN in TensorLayerX

alt tag

Prerequisites

🔥🔥🔥🔥🔥🔥 You need install TensorLayerX at first!

🔥🔥🔥🔥🔥🔥 Please install TensorLayerX via source

pip install tensorlayerx

or

pip install git+https://github.com/tensorlayer/tensorlayerx.git 

Usage

First, download the aligned face images from google or baidu to a data folder.

Please place dataset 'img_align_celeba.zip' under 'data/celebA/' by default.

Your directory structure should look like this:

dcgan/
    └── data.py
    └── model.py
    └── train.py
    └── README.md
    └── data
          └── celebA
            └── img_align_celeba.zip

Second, train the GAN:

$ python train.py

Result on celebA

<a href="http://tensorlayer.readthedocs.io"> <div align="center"> <img src="img/result.png" width="90%" height="90%"/> </div> </a>