search

GesilaA / deepsort_tensorrt

This is a TensorRT based deepsort project
77 stars 26 forks source link

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

Open Hezhexi2002 opened 2 years ago

Hezhexi2002 commented 2 years ago

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

GesilaA commented 2 years ago

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

Hezhexi2002 commented 2 years ago

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

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

opentld commented 2 years ago

您把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 commented 2 years ago

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