实现将DiscoGAN 原版例程image_translation https://github.com/SKTBrain/DiscoGAN
移植到matlab
- DiscoGAN.mlx 主程序
- download.mlx 下载训练集的脚本
- 不建议用这个下训练集 建议用迅雷下载 winrar解压 快很多
- Discriminator.m Generator.m 判别器和生成器的实现
- get-*loss.m 网络训练用的几种loss函数实现,与原版代码https://github.com/SKTBrain/DiscoGAN对应
- glorot.m 生成网络权重初始值用的
仅在Matlab 2019B上测试过
在神舟战神z7 kp7ec上测试,16分钟一个epoch
官方torch实现约18分钟一个epoch 相差不大