摘要
本发明提供了一种基于参数优化生成对抗网络的手写数字生成方法。包括:准备手写数字数据集作为样本训练数据集,采样得到真实数据并初始化随机噪声数据;建立生成对抗网络,初始化生成器网络权值参数θ和判别器网络权值参数ω;通过搬土距离W建立生成器损失函数和判别器损失函数,并且为判别器损失函数额外添加梯度惩罚损失项;迭代训练生成器网络和判别器网络,优化生成器网络权值参数θ和判别器网络权值参数ω。本发明实例解决了原始生成对抗网络收敛缓慢,训练不稳定,计算开销大等问题。实现了对生成对抗网络的优化,充分提高了生成对抗网络的网络性能,并且生成器能够生成质量更高的手写数字图像。
- 单位