win10+cudn9+anaconda4.6.11(python3.7)下pysot环境部署

PySOT 是由 SenseTime 视频情报研究团队设计的软件系统。它实现了最先进的单一对象跟踪算法,包括 SiamRPN 和 SiamMask。它是用 Python 编写的,由 PyTorch 深度学习框架提供支持。该项目还包含用于评估跟踪器的 Python 工具包端口。PySOT 的目标是为视觉跟踪研究提供高质量、高性能的代码库。它旨在灵活,以支持新颖研究的快速实施和评估。(向大神致敬

注:如果对于对于安装有疑问,可以先到github上将项目pull到本地,在pysot的根目录下有个文档“README.md”(图1),打开该文档,在该文档的"INSTALLATION"标题下有相关安装步骤的说明(图2)。                                                                                                                                                                 

                                                                                  图1                                                                                                                   

                                                     

                                                                                                图2

GitHub项目地址:https://github.com/STVIR/pysot

下面是我在win10上安装pysot的步骤和遇到的困难:

依赖环境:

  • Conda with Python 3.7.
  • Nvidia GPU. ---可根据自己的机器安装,我的机器显卡是GTX1060,我安装了conda9.0
  • PyTorch 0.4.1 ---最好与readme.md中的版本一致
  • yacs
  • pyyaml
  • matplotlib
  • tqdm
  • OpenCV

注:CUDN和Anaconda的安装在这里可以参考去看相关博客。

在安装之前先去https://github.com/STVIR/pysot将项目下载到本地。点击开始-》Anaconda3(64-bit)-》Anaconda Prompt。

一、创建pysot的工作环境,并激活环境

直接在Anaconda Prompt命令行输入以下命令即可:

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

如果python版本不清楚,可以在Anaconda Prompt命令行输入"python":

Python3.7.3就是python的版本信息(最好按教程安装python3.7)。

此处无坑,只要命令正确即可。如果出现下图表示内容表示已经激活pysot的工作环境。

二、安装numpy,opencv,pytorch(此处有坑)

 如果前面已经安装过上述工具则可以忽略,直接到下一步,后面有问题再返回执行本步骤。

1、安装numpy

conda install numpy

2、安装opencv

pip install opencv-python

注意,如果在Anaconda Prompt中执行出现超时(time out)的错误,试着到以管理员的身份运行dos(右键->Windows PowerShell(管理员)A),然后执行上面的命令,如果依然不成功,可能是conda的源有问题,需要重新配置源。我在安装时就在此处采坑,可以用我安装成功的源试着安装。

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
show_channel_urls: true

以上是我安装成功的源,直接在C:\Users\用户名目录下找到.condarc文件(该文件是隐藏文件,如果没找到先打开隐藏文件显示,然后在系统盘根目录下搜索).condarc,将.condarc文件备份后,直接打开用将上面的内容替换原内容即可,然后重新打开Anaconda Prompt,输入安装命令。

3、安装pytorch

在install中用到命令:

conda install pytorch=0.4.1 torchvision cuda90 -c pytorch

但是每次我在下载时下载到了3%就会出现错误:CondaHTTPError。。。,提示获取不到url,然后经过几个小时的搜索和不断的安装,中与用一下的命令安装成功:

conda install pytorch=0.4.1 torchvision cuda90 

由于一开始我没有加入conda外部的pytorch源- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/,后来才加入,使用-c命令时,会指定使用以前的源,所以导致在安装pytorch时,获取不到url,当去掉-c命令后,就会使用加入后的源去安装(大坑)。在单独用conda安装pytorch时,遇到此种情况也可以使用该方法。

三、安装其依赖工具

只需要在Anaconda Prompt输入以下命令:

pip install pyyaml yacs tqdm colorama matplotlib cython tensorboardX

此处几乎没有什么问题会出现。

四、生成扩展环境

找到pull到本地的项目,进入到pysot根目录下,执行:

python setup.py build_ext --inplace

如果没错误,就结束了了环境的部署。这一步几乎无坑。安装好之后我们就可以测试了,但是在测试之前我们还需要下载models 文件,model文件被poysot作者贴心的放到了百度云盘,云盘地址:https://pan.baidu.com/s/1GB9-aTtjG57SebraVoBfuQ#list/path=%2F,提取密码是 j9yb。将下载好的model.pth放到\pysot\experiments\siamrpn_r50_l234_dwxcorr下,因为我没有webcam,所以直接使用作者\pysot\demo下的demo视屏,将目录切换到pysot根目录下,输入:(此处有坑)

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

我们会发现,提示找不到pysot,那是因为python在默认的搜索路径中找不到pysot的路径,所以会报错,那么我们就将pysot的路径加入到python的搜索路径即可,在linux下输入:

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

在windows下就要调用python的sys.path.append(pysot的项目路径),首先进入python命令行:例如我的是

import sys
sys.path.append('D:\install\pysotinstall\pysot')

然后执行sys.path看看自己的路径有没有加进去。

 添加成功后重新执行测试命令,就可以看到现象了。  但是该方法每次都要在打开  Anaconda Prompt时导入一次pysot的路径,这样会显得特别麻烦,所以我们可以在系统路径中直接添加,后面就再也不用重新添加该命令了,具体做法是:右键我的电脑->属性->高级设置-> 环境变量->系统变量中新建  变量名为PYTHONPATH,变量值为pull下来代码的的目录(pysot的根目录),点击确定后就可以在sys.path中看到我们添加的路径了。                                                                                                              

在安装过程中,可能会遇到各种问题,但是最大的问题不是安装的步骤出现问题,而是在通过Anaconda的源去下载相应的软件包时,会出现获取不到url,或者timeout的问题,所以在安装anaconda时最好将源配置成国内清华,豆瓣的源,在一步步安装。

  • 2
    点赞
  • 29
    收藏
    觉得还不错? 一键收藏
  • 48
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 48
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值