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

关于与tensorrrx对接的相关问题? #14

Open
Hezhexi2002 opened this issue Nov 9, 2021 · 4 comments
Open

关于与tensorrrx对接的相关问题? #14

Hezhexi2002 opened this issue Nov 9, 2021 · 4 comments

Comments

@Hezhexi2002
Copy link

大佬,我是中北大学的一名大二学生,在我们学校rm战队里负责用神经网络识别装甲板,目前我已经实现了tensorrtx加速yolov5,所以我想问一下该如何将deepsort加入我的工程中,而且怎样去track我自定义的类别呢

@GesilaA
Copy link
Owner

GesilaA commented Nov 9, 2021

可以参考demo.cpp,就是将yolov5的检测结果使用vector进行封装,然后调用DeepSort的sort(cv::Mat, vector)方法就可以了;
追踪特定类别其实就是:只将需要追踪的类别丢到vector里面
这个工程有点乱了哈。。一直也没时间整理一下

@Hezhexi2002
Copy link
Author

可以参考demo.cpp,就是将yolov5的检测结果使用vector进行封装,然后调用DeepSort的sort(cv::Mat, vector)方法就可以了;
追踪特定类别其实就是:只将需要追踪的类别丢到vector里面
这个工程有点乱了哈。。一直也没时间整理一下

tensorrtx中的返回结果就已经是一个vector了,所以我再好好研究一下您的代码,到时候试一试,对了,大佬,如果方便的话能够加个微信交流吗,就是我之后遇到问题好向您请教

@opentld
Copy link

opentld commented May 25, 2022

您把track.txt里的数据读进去进行sort,我看你读进去的数据就已经有ID和confidence了,那么,这个confidence是我检测出的值?还是track的confidence?这个目标ID我如过detect出来了,那跟踪的意义何在呢?
我理解,我应该只把检测出的box的x1 y1 x2 y2扔进去,你的sort应该给我返回ID和confidence才对。但我看你DEMO并不是这么做的。而且。我尝试把detection的检测结果丢进去,如果我不给ID的话,你sort返回的ID永远是-1。
但如果我给ID的话,跟踪的意义何在?我都已经有ID了啊??
@GesilaA

@GesilaA
Copy link
Owner

GesilaA commented May 26, 2022

@opentld confidence是检测输出的结果,跟踪是不会输出置信度的,因为一个目标后面不会变成多个目标,所以是取top1。track.txt里面的ID是类别id,不是跟踪id。算法输出的类别id和置信度与检测是一致的,它会生成跟踪id

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

3 participants