生成 RGB 图像

你好,我想问一下 SPADE 是否采用固定的维度映射,因为我正在尝试使用您的预训练模型来生成 COCO Stuff 数据集的 RGB 图像,但我遇到了这个错误: 我们遇到了一个错误,抱怨维度与 SPADE 使用的输入不匹配。 RuntimeError: 给定 groups=1,大小为 [1024, 184, 3, 3] 的权重,预期输入 [1, 186, 8, 8] 具有 184 个通道,但实际得到 186 个通道。

我使用的语义映射的维度是 256*256。我想了解 SPADE 用于生成 RGB 图像的语义映射的属性。如果您能推荐一些解决此问题的方法,我将不胜感激。