Keeping aspect ratio with CropAndResize Plugin

[MaximeDebarbat]

Hi,

I'm looking for a way to keep the aspect ration with the plugin CropAndResize (yolo style):
https://github.com/NVIDIA/TensorRT/blob/release/10.6/plugin/cropAndResizePlugin/cropAndResizePlugin.cpp

I tried to create a custom torch model which would do this padding but it's unsuccessful for now.
Any help would be greatly appreciated!

Cheers

[lix19937]

You can ref tf.image.crop_and_resize.

[asfiyab-nvidia]

cc @venkywonka for additional comment

[venkywonka]

Hi @MaximeDebarbat , thank you for taking the time to create the github issue. Could you provide more details on which specific part of the plugin you're facing problem with?
Is the plugin now working as expected in your setup?

I tried to create a custom torch model which would do this padding but it's unsuccessful for now.

I'm afraid I don't understand how this might be related to the plugin. You mean the above was the alternative you tried?
Thank you @lix19937 for help. @MaximeDebarbat have you tried @lix19937 's suggestion?

Thanks!