Accept input image with variable size?

[chunchet-ng]

HI there, how do i change the code so that i can accept input training image of variable size? And when i want to evaluate the model, it should be able to output masked testing image of variable size too?