admin 管理员组文章数量: 1184232
源码地址:https://github/ifzhang/ByteTrack
我的cuda版本为10.2,先安装pytorch
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch/whl/torch_stable.html
下载源码
git clone https://github/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:
安装pycocotools
pip install cython
安装cython_bbox
使用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视频流进行实时识别,但同局域网下延迟却很大,不知原因为何,正在解决!
参考:
解决OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.报错问题_码农小C的博客-CSDN博客_libiomp5md.dll
Windows上遇到错误之No module named 'cython_bbox'_慕课手记 (imooc)
多目标跟踪评价指标 - 知乎 (zhihu)
目标跟踪之 MOT 经典算法:ByteTrack 算法原理以及多类别跟踪_kuweicai的博客-CSDN博客_多类别多目标跟踪
版权声明:本文标题:Windows搭建ByteTrack多目标跟踪算法 内容由网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:http://www.roclinux.cn/b/1764644419a3304887.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论