How to modify the data representation to support different sizes of layouts?

[mln00b]

Hi, Currently the PubLayNet example generates layouts of size 256x256, a fixed square aspect ratio.
How do I modify the representation to make it support different aspect ratios during inference?