Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Does this support yolov5 4.0 release? #1

Open
KevinZhanggg opened this issue Feb 17, 2021 · 5 comments
Open

Does this support yolov5 4.0 release? #1

KevinZhanggg opened this issue Feb 17, 2021 · 5 comments

Comments

@KevinZhanggg
Copy link

No description provided.

@positive666
Copy link
Owner

Yes support, 4.0 release should require torch 1.7. The conversion model provides two project conversion methods under trans_project. Recently, I will supplement this project when I have time. You can try to convert

@KevinZhanggg
Copy link
Author

@positive666 我用tensorrtx 那个转换出来的的engine 可以推理出来,输出的结果也很正常。 但是放在deepstream 里之后推理就有很多boxes,很多误检。 你有遇到这个情况吗?

@positive666
Copy link
Owner

之前没遇到过,但你说TensorRt没问题前提下,这里可能有两个具体原因?是否你的deepstream下的 解析cpp代码中NMS有问题 ;或者是阈值问题 你可以调一下看看,我在第一次作V3的模型转换时候遇到过,类似问题,当时是Nms的阈值造成的

@KevinZhanggg
Copy link
Author

@positive666 我感觉是NMS的问题,但没先找出来具体那里需要修改,阈值大小我修改过,还是一样。冒昧的问下,可以在给个微信或者qq吗,方便的话,想讨论下。谢谢🙏

@positive666
Copy link
Owner

这个问题 就是tensorrt生成引擎文件不能在当前目录下生成 要使用非根目录

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants