[pix2pixHD] 在 png 图像训练模型中生成的灰度图像

你好,我在训练 png 图像时遇到了 pix2pixHD 模型的问题。 我稍微修改了一下代码,以便正常训练,例如正则化部分和 --no_vgg_loss 选项。 但是,在生成图像时,它只呈现为灰度图像。 checkpoint 文件夹中的每个图像都是灰度图像,包括 input_label、real_image 等。 我发现另一个仓库中有人遇到了同样的问题,但尚未解决。 是否有谁能解决这个问题呢? --- 我修改的代码 在 ./data/aligned_dataset 中,第 42/51/63 行 ... convert('RGB') -> convert('RGBA') 在 ./data/base_dataset.py 中,第 56、57、61 行 ... [transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))] -> [transforms.Normalize((0.5, 0.5, 0.5, 0.5), (0.5, 0.5, 0.5, 0.5))] 在 --nc_input 4 --nc_output 4 --no_vgg_loss --no_instance 选项下进行训练