NVIDIA-AI-IOT / trt_pose

Real-time pose estimation accelerated with NVIDIA TensorRT
MIT License
974 stars 291 forks source link

Training non-square aspect ratios #148

Open jpapon opened 2 years ago

jpapon commented 2 years ago

Hello,

I've managed to get training up and running without issue as long as I keep image_shape square. If I try to change it something like [480,272] I get an error like the following: RuntimeError: The size of tensor a (128) must match the size of tensor b (64) at non-singleton dimension 2

Is it possible to train on non-square aspect ratios, and if so, what parameters do I need to adjust to do this? If not, what's the recommended way of using the network on input data that isn't square? Should I crop images to square, or does the network expect squished people because of training and I should just pass it a squished image?

Thanks!