Advertisement

YOLOX+Deepsort 实现多目标追踪

阅读量:

项目简介:

基于YOLOX与Deepsort的结合方案完成车辆与行人的追踪与计数功能,并将其代码打包成一个Detector类形式,在嵌入到个人项目中时更加方便地实现了功能。

代码地址(欢迎star):

https://github.com/Sharpiless/yolox-deepsort

最终效果:

在这里插入图片描述

运行demo:

复制代码
    python demo.py

训练自己的模型:

训练好后放到 weights 文件夹下

调用接口:

创建检测器:

复制代码
    from AIDetector_pytorch import Detector
    
    det = Detector()

调用检测接口:

复制代码
    result = det.feedCap(im)

其中 im 为 BGR 图像

返回的 result 是字典,result[‘frame’] 返回可视化后的图像

联系作者:

B站:https://space.bilibili.com/470550823

:<>

AI Studio:https://aistudio.baidu.com/aistudio/personalcenter/thirdview/67156

Github:https://github.com/Sharpiless

全部评论 (0)

还没有任何评论哟~