作者chengreg (想重回校园的工程师)
看板DataScience
标题[问题] CycleGAN horse2zebra 训练失败求助
时间Thu Jun 8 10:38:02 2023
作业系统: Win10
问题类别: Keras CycleGAN 训练失败
使用工具: Python 3.7(64 bit), tenorflow(2.3.0), Keras(2.4.0)
问题内容: 练习 CycleGAN 训练 horse2z 失败
各位前辈好, 小弟学习CycleGAN 利用书上
(
https://www.books.com.tw/products/0010868212)
的范例进行训练与练习
然而CycleGAN的范例 如下 :
https://github.com/eriklindernoren/Keras-GAN/tree/master/cyclegan
小弟在练习 apple2orange 进行训练时, 都有成功转换风格
然而转换成 horse2zebra 训练至 epochs 100 -> 200 次, 皆无法成功转换风格
其 g-loss 在 5.X 左右 d-loss 已是 0.1左右了
小弟实验变化其参数 :
1.於 discriminator network 增加 Dropout 0.2 - 0.5 反覆实验, 结果无法成功
2.於 generator network 增加 Dropout 0.2 - 0.5 反覆实验, 结果无法成功
3.於 discriminator network 增加 GaussianNoise layer 赋予 0.01, 结果无法成功
4.於 generator network 增加 GaussianNoise layer 赋予 0.01, 结果无法成功
5.调整 optimizer 采用 RMSProp 结果失败
6.调整 神经元数目 (self.gf 与 self.df) 增加一倍, 实验失败
以上状况, 目前皆无法成功将 hosre2zebra 但 apple2orange 皆成功
不知前辈是否有此经验,不吝惜指导小弟
该范例问题之症结点与方向该如何突破
谢谢各位前辈
--
※ 发信站: 批踢踢实业坊(ptt.cc), 来自: 1.164.241.158 (台湾)
※ 文章网址: https://webptt.com/cn.aspx?n=bbs/DataScience/M.1686191884.A.5B5.html
1F:推 chang1248w: 两个范例的资料集大小分别多少? 06/10 10:56
2F:→ chang1248w: 跟dropout比,不如增加data augmentation 的强度 06/10 10:57
3F:推 jack1218: Cycle GAN是真的不好用 06/11 10:21