About inference processing

[nldhuyen0047]

Hello, Thanks for your excellent work. I have a question about the inference processing. When I inference, the size of the input images resized 256x256 at the result. I would like to ask that If I want to keep the size of the input image with the result, what should I do?

Thank you so much.