conv-neural-network - 生成图像 CycleGAN 上的白点

Question

我正在尝试实施cyclegan。但是，即使在 10 或 25 个 epoch 之后，我生成的图像上似乎总是会出现白点。我想知道可能出了什么问题？我应该继续训练，问题就会消失吗？或者有任何关于如何解决这个问题的提示？

图片

标签： conv-neural-networkgenerative-adversarial-network

Answer 1

白点是在绘制图像时剪裁过大的模型输出值的结果。

在matplotlib.imshow()的文档中它说：

(...) 具有 RGB 值（0-1 float 或 0-255 int）的图像 (...) 超出范围的 RGB(A) 值被剪裁。

在不了解您的架构的情况下，我会：

1. 查看模型中的激活函数和生成器的最终激活。
2. 检查您的损失函数，您的损失目标可能偏爱高输出值
3. 尝试训练更多 epoch，可能模型学会了自己避免裁剪。