python - 从经过灰度训练的 GAN pkl 文件生成图像

Question

我已经使用一组灰度图像训练了一个 stylegan，现在我被困在 generate.py 中，因为它全部用于生成 RGB 图像。我收到此错误：

ValueError: not enough image data

当我将 generate.py 存档从 stylegan pytorch 从 RGB 更改为 L 时：

ValueError: Too many dimensions: 3 > 2.

这是整个 generate.py 代码： https ://github.com/mit-han-lab/data-efficient-gans/blob/master/DiffAugment-stylegan2-pytorch/generate.py

我已经在互联网上搜索了几个小时并尝试了很多东西，但是我对 python 还很陌生，所以我已经没有什么想法需要在 generate.py 中进行更改才能获得图像，有人可以帮忙吗我？

Answer 1

我也经常遇到这个问题...

最后发现它与索引img var有关。

在此处检查第 137-140 行：https ://github.com/pgtinsley/stylegan3/blob/main/gen_images_gray.py

这适用于 StyleGAN3，但代码/逻辑应该很容易转移到 SG2。

祝你好运！