该项目提供了一个使用PyTorch实现的BEGAN(Batch Generative Adversarial Networks)框架。虽然目前仍在开发中，但提供了下载CelebA数据集的脚本和训练模型的说明。要训练模型，需要安装必要的依赖并运行指定脚本。尽管BEGAN-pytorch在生成人类面部图像方面表现不如BEGAN-tensorflow，但两者都存在模式崩溃问题，可能与学习率调度有关。
BEGAN in PyTorch
This project is still in progress. If you are looking for the working code, use BEGAN-tensorflow.
Requirements
Python 2.7
requests (Only used for downloading CelebA dataset)
TensorFlow (Only used TensorBoard for logging)
Usage
First download CelebA datasets with:
$ apt-get install p7zip-full # ubuntu
$ brew install p7zip # Mac
$ python download.py
or you can use your own dataset by placing images like:
data
└── YOUR_DATASET_NAME
├── xxx.jpg (name doesn't matter)
├── yyy.jpg
└── ...
To train a model:
$ python main.py --dataset=CelebA --num_gpu=1
$ python main.py --dataset=YOUR_DATASET_NAME --num_gpu=4 --use_tensorboard=True
To test a model (use your load_path):
$ python main.py --dataset=CelebA --load_path=./logs/CelebA_0405_124806 --num_gpu=0 --is_train=False --split valid
Results
BEGAN-tensorflow at least can generate human faces but BEGAN-pytorch can't.
Both BEGAN-tensorflow and BEGAN-pytorch shows modal collapses and I guess this is due to a wrong scheuduling of lr (Paper mentioned that simply reducing the lr was sufficient to avoid them).
(in progress)
Author
Taehoon Kim / @carpedm20
扫一扫
1822
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
