V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
爱意满满的作品展示区。
xieincz
V2EX  ›  分享创造

分享最近做的一个关于在线多人追踪的项目

  •  
  •   xieincz ·
    xieincz · 2023-01-15 20:10:42 +08:00 · 1064 次点击
    这是一个创建于 465 天前的主题,其中的信息可能已经有所发展或是发生改变。

    检测器采用的是 YOLOv5 (可以轻松替换成其他同类的检测器),追踪器采用的是 ByteTrack 。和其他同类项目不同的是加入了多线程处理以及用 C++和 TensorRT 加快推理速度。一个线程负责读取来自视频文件(可以轻松更改为摄像头)的帧,一个线程负责用 YOLOv5 得到检测框,一个线程用 ByteTrack 给各个检测框 reid ,还有一个线程负责将结果绘制到视频文件中。在多 CPU 核心(>=3 核)的设备上的效果比目前其他同类的项目要更快。而且还可以根据需要给各个线程设置 CPU 亲和性(将某线程绑定到某个 CPU 核心,该功能仅限于 Linux 平台)。

    此外还利用 SWIG 包装了接口,方便在 python 中像调用一个库一样使用本项目。

    欢迎大家来试用 :)

    Github: YOLOv5_ByteTrack_Multithreading_TensorRT

    hifans
        1
    hifans  
       2023-01-16 11:16:45 +08:00
    @xieincz ,请问有兴趣合作没?
    LXGMAX
        2
    LXGMAX  
       2023-01-16 15:40:03 +08:00
    玩这么大?
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   我们的愿景   ·   实用小工具   ·   3400 人在线   最高记录 6543   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 28ms · UTC 11:36 · PVG 19:36 · LAX 04:36 · JFK 07:36
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.