deepsort_tensorrt
deepsort_tensorrt copied to clipboard
关于与tensorrrx对接的相关问题?
大佬,我是中北大学的一名大二学生,在我们学校rm战队里负责用神经网络识别装甲板,目前我已经实现了tensorrtx加速yolov5,所以我想问一下该如何将deepsort加入我的工程中,而且怎样去track我自定义的类别呢
可以参考demo.cpp,就是将yolov5的检测结果使用vector<DetectBox>进行封装,然后调用DeepSort的sort(cv::Mat, vector<DetectBox>)方法就可以了; 追踪特定类别其实就是:只将需要追踪的类别丢到vector<DetectBox>里面 这个工程有点乱了哈。。一直也没时间整理一下
可以参考demo.cpp,就是将yolov5的检测结果使用vector<DetectBox>进行封装,然后调用DeepSort的sort(cv::Mat, vector<DetectBox>)方法就可以了; 追踪特定类别其实就是:只将需要追踪的类别丢到vector<DetectBox>里面 这个工程有点乱了哈。。一直也没时间整理一下
tensorrtx中的返回结果就已经是一个vector了,所以我再好好研究一下您的代码,到时候试一试,对了,大佬,如果方便的话能够加个微信交流吗,就是我之后遇到问题好向您请教
您把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
@opentld confidence是检测输出的结果,跟踪是不会输出置信度的,因为一个目标后面不会变成多个目标,所以是取top1。track.txt里面的ID是类别id,不是跟踪id。算法输出的类别id和置信度与检测是一致的,它会生成跟踪id