CVAE-GAN_tensorlayer
具有tensorlayer的CVAE-GAN实现。
例子
训练VAE输出
测试VAE输出和侦察输出
依附
张量流张量层麻木进度条2
我的测试环境是tensorflow-gpu-1.10,tensorlayer-1.91,gtx970m-3g。
一些问题和注意
emmm ...此实现可能与页面有所不同。 我尝试了WGAN-GP和LS-GAN的丢失,但是效果不好。 也许我的代码是错误的。 然后我将鉴别器变成自动编码器并尝试BEGAN的损失,这看起来不错。 训练后的VAE输出仍然模糊,但是我发现鉴别器的重建图像可以使VAE输出更清晰。 主要的网络结构是我的简单自定义resnet,它应与DCGAN不同。
训练日志
我的训练过程输出和日志可以分别在imgs和logs文件夹中找到。 训练日志已被压缩,仅需解压缩。
如何使用
测试
python3 test.
1