How to get the same output size as the input image?

[nass2020]

The algorithm works very well, but how to get the same output size as the input image? All output images are (480x368)?
Thanks a lot and congratulations for this great work

[HelixNGC7293]

I figured this out!
You can change the function resize_368_short_side(img) in VP_code\data\dataset.py, change all the 368 value inside this function to your image's actual width/height

this number of output image size needs to be a multiple of 16, otherwise it will cause a grid & input size mismatch error
You can check out my result at here: #10 (comment)