ATOM追踪代码在Windows上运行

本文详细记录了在Windows上运行ATOM物体追踪代码的全过程,包括安装anaconda、VS2015/2017、CUDA、cudnn、prpooling等依赖,解决ninja、pycocoapi的编译问题,以及torch.load()的CUDA版本兼容性调整,最终成功运行代码的步骤和遇到的挑战。
摘要由CSDN通过智能技术生成

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

说明如下
Github原文还有一个小问题要注意吧,大家要看好自己的显卡支持的最高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 <
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值