RK3399Pro 如何使用deepsort

[复制链接] · 发表于 2019-12-16 09:14:17

请教各位论坛里面的大神有谁知道如何在RK3399Pro里面使用deeposrt的吗？我想要实现车辆跟踪的功能，需要将yolov3和deepsort相互结合起来，像这个项目一样：https://github.com/Qidian213/deep_sort_yolov3。
希望可以和各位探讨一下。

只看该作者 · 发表于 2019-12-20 14:47:33

你只需要转换下yolov3模型就好了,其他的(比较向量的余弦距离,匈牙利匹配算法,kalman跟踪都可以沿用呀),没难点在.
自己试试就好了.
另外,yolo3转换模型,置顶帖子里有demo.

只看该作者 · 发表于 2019-12-21 10:51:54

xiaqing10 发表于 2019-12-20 14:47
你只需要转换下yolov3模型就好了,其他的(比较向量的余弦距离,匈牙利匹配算法,kalman跟踪都可以沿用呀),没难 ...

但是deepsort还使用了mars-small128.pb这个tf模型，也需要转模型，但是有rknn不支持的层

只看该作者 · 发表于 2019-12-23 10:35:09

YEY 发表于 2019-12-21 10:51
但是deepsort还使用了mars-small128.pb这个tf模型，也需要转模型，但是有rknn不支持的层 ...

他只是一个reid模型. 不支持的话,看哪些层不支持,或者再随表找个训练好的reid就可以了. 甚至于你可以直接用Hog等颜色特征先代替用着,跑起来了再换模型.

只看该作者 · 发表于 2020-5-8 11:15:06

大佬，想问下在3399pro上跑起来deepsort了嘛