本项目提供了一个基于TensorFlow框架的WGAN-GP( Wasserstein GAN with Gradient Penalty)的实现方案。它能够有效地训练生成对抗网络,并应用于图像生成等任务中。
WGAN-GP-张量流存储库提供了一个使用Tensorflow实现的代码库,适用于MNIST、CIFAR-10以及ImageNet64数据集。除README.md文件中第一行图像外的所有样本均由神经网络生成。
安装先决条件:
- Python 3.5, 3.6 或 3.7
- python3-tk
对于Ubuntu或Debian系统,请使用以下命令进行安装:
```
sudo apt install python3.5 python3.5-tk
```
创建虚拟环境:
在视窗中,运行如下命令来激活虚拟环境:
```
venv/Scripts/activate
```
或者在重击(Linux)环境中执行如下命令以激活它:
```
source venv/bin/activate
```
安装虚拟环境中的依赖项,请使用以下命令:
```
pip install -r requirements.d/venv.txt
```
创建运行时环境,可以使用tox工具来完成。要仅用CPU的Tensorflow进行安装,请执行下面这个命令:
```
tox --notest
```
若需要在Nvidia GPU上运行,则请根据相关文档配置GPU支持。