多目标跟踪算法,使用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

 

你可能感兴趣的:(深度学习)