DeepStream-Yolo icon indicating copy to clipboard operation
DeepStream-Yolo copied to clipboard

Do Inference size width and height have to be equal?

Open Xinxinatg opened this issue 2 years ago • 1 comments

Screenshot from 2022-09-22 14-05-45 I trained the yolov5 model with size 640, and wants to set the inference size to be as big as possible. In one case the width is 1920 and the height is 1080 which are the same as the size of the input images. But I was given the error as indicated in the screenshot. Also, when the model was working where the inference width and height are both 1920, the performance is significantly lower than what is like in the orignial torch model. Really appreciate it if someone can offer some insight into it!

Xinxinatg avatar Sep 22 '22 04:09 Xinxinatg

It doesn't need to be width=height but you must use the same size of the trained model (640) to get the correct output. The sizes must be multiple of 32.

marcoslucianops avatar Sep 23 '22 12:09 marcoslucianops

It doesn't need to be width=height but you must use the same size of the trained model (640) to get the correct output. The sizes must be multiple of 32.

thanks, this helps!

Xinxinatg avatar Sep 29 '22 06:09 Xinxinatg