Skip to content

Latest commit

 

History

History
27 lines (15 loc) · 696 Bytes

说明.md

File metadata and controls

27 lines (15 loc) · 696 Bytes

DiscoGAN-Matlab

实现将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 相差不大