Support image resize

[WikiChao]

Hi, thank you for the great work!

When I run the computation, I get this error

RuntimeError: stack expects each tensor to be equal size, but got [3, 426, 640] at entry 0 and [3, 640, 586] at entry 1

It seems all images need to be resized to the same shape before batch processing. Can you include this in the code?

Thanks

[loki-veera]

Hello @WikiChao,
Thank you for bringing this up.
You are absolutely right, the error is due to the input images having different dimensions.
We are aware of and working on this issue, and will incorporate it in future releases. In the meantime, as a workaround, we recommend resizing the images beforehand.

Thanks.