About the size of output image of generator

[血小板自动机]

I want to train pix2pix coloring model with 256 * 256 pictures, and then I can color pictures of any size. How can I make the generator output pictures of any size?

[Jun-Yan Zhu]

It depends on the generator architecture. The UNet generator used in the original pix2pix only supports images with height and width divisible by 256. You can try other kinds of architecture. For example, the Resnet-based generator supports images with height and width divisible by 4.