多目標跟蹤算法,使用pytorch實現RFBNet+deepsort,基於RFBNet的多目標跟蹤算法

       由於deepsort使用的faster-rcnn速度很慢,無法達到實時跟蹤。所以在deepsort框架的基礎上使用RFBNet來代替faster-rcnn。檢測速度在GTX1060上可以達到15-20FPS。

       由於檢測算法在訓練過程中數據集比較少,訓練的模型並不是很好。所以在多目標跟蹤過程中會出現id丟失或者跟錯的現象。

demo演示:https://www.bilibili.com/video/av62427886/

github:https://github.com/ziweizhan/deepsort-RFBNet

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章