Is there some way to preserve image size after drawing?

[Yikai-Wang]

Hi there,
Thanks for your excellent work! I am trying to follow your work and when I want to generate line drawings with different input size, the output image cannot preserve the size.

For example, input size of (390, 500, 3) leads to the output of (392, 500, 3), and input size of (300, 230, 3) leads to the output of (300, 232, 3).

The difference is small but still make some problem to match inputs and outputs in pixels. Is there some way to generate a drawing with the same size of input image?