PaddleGAN icon indicating copy to clipboard operation
PaddleGAN copied to clipboard

stylegan_v2_256是否可以训练生成512或者1024尺寸的图片,如何配置yaml?

Open zhw511006 opened this issue 11 months ago • 2 comments

在config文件夹里面找到了stylegan_v2_256_ffhq.yaml配置, model: name: StyleGAN2Model generator: name: StyleGANv2Generator size: 256 style_dim: 512 n_mlp: 8 discriminator: name: StyleGANv2Discriminator size: 256

这样训练完的模型可以生成尺寸为256的图片,简单修改其中size为512或者1024,进行训练的时候直接报错,如下:

ValueError: (InvalidArgument) Input(Y) has error dim.Y'dims[0] must be equal to 12800But received Y'dims[0] is 8192 [Hint: Expected y_dims[y_ndim - 2] == K, but received y_dims[y_ndim - 2]:8192 != K:12800.] (at /paddle/paddle/phi/kernels/impl/matmul_kernel_impl.h:315) [operator < matmul_v2 > error]

请问如何修改yaml配置可以生成512或者1024尺寸的图片,谢谢!

zhw511006 avatar Mar 01 '24 08:03 zhw511006

在组网阶段一些输入shape也需要检查下

jerrywgz avatar Mar 04 '24 09:03 jerrywgz

有没有一种可能把大图切成多个256小图,生成后再拼接回去。就是不知道会不会有接痕?

1196228233 avatar Jun 01 '24 12:06 1196228233