PySOT 安装与模型测试

最新推荐文章于 2024-08-08 08:11:43 发布

不会飞的鹰08

最新推荐文章于 2024-08-08 08:11:43 发布

阅读量1.1k

点赞数

分类专栏：目标跟踪文章标签：深度学习

本文链接：https://blog.csdn.net/weixin_41449637/article/details/109071383

版权

目标跟踪专栏收录该内容

12 篇文章 2 订阅

订阅专栏

该博客详细介绍了如何安装和配置PySOT目标跟踪框架。首先，通过conda创建环境并安装相关依赖，包括PyTorch和OpenCV等。接着，下载工程文件和模型，并设置环境变量。然后，在PyCharm中配置运行参数，运行demo.py，实现目标跟踪。最后，展示了运行效果。

摘要由CSDN通过智能技术生成

安装步骤

-下载工程文件

https://github.com/STVIR/pysot.git

- 创建conda环境

https://github.com/STVIR/pysot/blob/master/INSTALL.md

采用第二种安装方式，手动配置环境,命令如下，自行添加清华源 https://pypi.tuna.tsinghua.edu.cn/simple

conda create --name pysot python=3.7
conda activate pysot

conda install numpy
conda install pytorch=0.4.1 torchvision cuda90 -c pytorch
pip install opencv-python

pip install pyyaml yacs tqdm colorama matplotlib cython tensorboardX

# change dir to project

python setup.py build_ext --inplace

下载模型文件

将百度云https://github.com/STVIR/pysot/blob/master/MODEL_ZOO.md 的文件下载下来，只复制pysot/experiments文件中的mode.pth,不替换相应的congfig.yaml

配置环境变量

export PYTHONPATH=/path/to/pysot:$PYTHONPATH

改为 export PYTHONPATH=/home/cody/PycharmProjects/pysot:$PYTHONPATH

或者使用pycharm 配置

https://blog.csdn.net/sements/article/details/105495812/

demo 运行

参考https://blog.csdn.net/sements/article/details/105495812/

这里使用pycharm运行，选择相应的Python 编译器，在run -> Edit Configurations，设置并选择运行的demo.py文件，在paramters中添加运行的参数

--config ../experiments/siamrpn_r50_l234_dwxcorr/config.yaml
--snapshot ../experiments/siamrpn_r50_l234_dwxcorr/model.pth
--video ../demo/bag.avi

效果展示

运行demo.py，弹出显示框，鼠标左键选中区域后回车即可