DCGAN in TensorLayerX
-
This is an implementation of Deep Convolutional Generative Adversarial Networks. Looking for Text to Image Synthesis? click here
-
For earlier version, please check dcgan release and tensorlayer
-
For more computer vision applications, check TLXCV
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