ATOM追踪代码在Windows上运行
ATOM追踪代码在Windows上运行
前言
本人刚刚本科毕业,准备出国所以跟着清华导师做个项目,最近开了物体追踪的坑,老师话不多说直接让跑代码,作为一个萌新,尤其是懒得装双系统的萌新,决定直接在windows上跑。从代码架构上可以看出作者很有代码功底,然而唯一不足的是他引用了适用性比较坑的prpooling代码,导致在Windows上跑炸裂。事先说明,如果实验室用linux环境的同学尽量用linux跑,windows跑要做好折腾的准备。既然你已经准备好啦,那就来吧
Github代码:https://github.com/visionml/pytracking
论文地址: https://arxiv.org/pdf/1811.07628.pdf
运行条件
- anaconda 自己去下一个装上就可以了
- vs2015/vs2017 对,你没有看错,真的需要这个东西。因为pycocoapi和ninja需要vs编译,不需要完整的,只需要VC部分就可以惹。因为我本科生当过C语言助教,所以电脑里有两个版本,导致路径和只有一个版本的路径不一样,所以后续配置环境变量的时候大家看好自己的路径
- VC别忘了装Windows Kit,如果我没记错的话2015是不带的
- cuda和cudnn,什么你没有cuda,那放弃吧,你跑不起来的,因为prpooling这部分只支持CUDA,CPU版本还没有实现。因此,用虚拟机也不可能跑起来的。
PrPooing的Github
https://github.com/vacancy/PreciseRoIPooling
说明如下
还有一个小问题要注意吧,大家要看好自己的显卡支持的最高CUDA版本,方法为 桌面右键-NVIDIA选项-帮助-系统信息-组件-NVCUDA.dll
- pytorch 看好自己的的cuda版本,去官网上下就好了,最好用anaconda,在自己的虚拟环境里安装好
github
作者在代码里引用了其他github代码,默认是不下载的,所以在download的时候别忘了进入pytracking/ltr/external/ 下载prpooling的源码
如果你用git下载需要两个命令
git clone https://github.com/visionml/pytracking.git
git submodule update --init
不然真正跑代码的时候会提示缺少包
安装各式各样的包
这一步还算简单,用记事本打开install.sh,打开anaconda prompt一步一步敲就好,创建虚拟环境就自己弄吧~我们从虚拟环境activate后开始
conda <