YOLOv5-LibTorch issues

The torchscript weight from YoLoV5 is not workable

7

Thanks very much for creating this repo. Now I am facing one problem that the converted torchscript weight from YoLoV5 is not working with the code you provided. The conversion...

Jelly123456

Error in inference on other size except (640,640)

3

Hi, I want inference my ```.torchscript.pt``` which is trained by this command and switch: ``` python train.py --img 128 128 --rect ... ``` but, by change resize function in this...

Yasin40

How to extract the raw results without using the func non_max_suppression?

First of all , thanks for you share. I'd like to know how to extract the raw results after the forward step? because i need to debug for the middle...

taotaoyuhust

memory error

Hi, I've tried your code on your provided weights its working fine, but when I tried my custom weight exported using ultralytics repo, even using older tags is not working....

shubham-mishra-verificient