Input an image of any size

[hanhan0521]

I am very interested in your work. But I have a question, can the entered image only say 512×512. I want to import an image of any size, what do I do?

[yuvraj108c]

hi, you have to resize the image before passing it into the model! As mentioned, this repo doesn't do any pre/post processing, only inference.

[yuvraj108c]

you can also export a onnx with different sizes using the export_onnx.py, then build the tensorrt engine from it