Thanks for your code contribution. Do I have to resize the picture to 256 when I use your pretrained checkpoint for the first stage of inference? My image size is 512*512 or other size, what should I do if I want to reconstruction it to the original size?
