Windows搭建ByteTrack多目标跟踪算法

最新推荐文章于 2024-09-04 01:17:21 发布

盛世芳华

最新推荐文章于 2024-09-04 01:17:21 发布

阅读量3k

点赞数 3

分类专栏：多目标跟踪文章标签： windows 目标跟踪 python

本文链接：https://blog.csdn.net/lanyan90/article/details/125915117

版权

2 篇文章 1 订阅

订阅专栏

我的cuda版本为10.2，先安装pytorch

pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

git clone https://github.com/ifzhang/ByteTrack.git
cd ByteTrack
pip install -r requirements.txt
python setup.py develop

pip install -r requirements.txt过程中会把我的torch、torchvision自动卸掉，重新安装一次pytorch！

执行python setup.py develop报错UnicodeDecodeError: 'gbk' codec can't decode byte

编辑setup.py，将

with open("README.md", "r") as f:

改为：

with open("README.md", "r",encoding='utf-8') as f:

pip install cython

使用pip install cython_bbox安装会报错

从官网下载cython_bbox库，在Windows本地进行编译，具体步骤如下：

①首先从官网下载cython_bbox库，进入官网后，单击Download files，再单击cython_bbox-0.1.3.tar.gz开始下载。下载地址：cython-bbox · PyPI

②将下载的cython_bbox-0.1.3.tar.gz解压，解压后打开文件夹，右击setup.py以记事本的方式打开，将第31行

extra_compile_args=['-Wno-cpp']

替换为：

extra_compile_args={'gcc': ['/Qstd=c99']}

然后保存。

③以命令行的方式进入cython_bbox-0.1.3目录，执行命令：

python setup.py build_ext install

下载模型

下载模型，并放于pretrained目录下，若文件夹不存在，则新建一个！

python tools/demo_track.py video -f exps/example/mot/yolox_x_mix_det.py -c pretrained/bytetrack_x_mot17.pth.tar --fp16 --fuse --save_result

如果出现Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.报错问题，编辑demo_track.py文件，增加以下代码：

import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"

若没有报错，则可在YOLOX_outputs目录下找到生成视频！

python tools/demo_track.py webcam -f exps/example/mot/yolox_m_mix_det.py -c pretrained/bytetrack_m_mot17.pth.tar --fp16 --fuse --save_result

建议选择bytetrack_m_mot17.pth.tar模型，基本0延迟！

尝试直接调用海康威视摄像头的rtsp视频流进行实时识别，但同局域网下延迟却很大，不知原因为何，正在解决！

参考：

关注

专栏目录