Win10 使用OWDETR
源代码链接
GitHub - akshitac8/OW-DETR: [CVPR 2022] Official Pytorch code for OW-DETR: Open-world Detection Transformer
原论文链接
https://link.zhihu.com/?target=https%3A//arxiv.org/abs/2112.01513
大佬的介绍链接
OW-DETR | 基于 Transformer 的开放世界目标检测器 - 知乎
下面是使用教程,根据readme的文字,在安装中要求ubuntu16,CUDA10.2,GCC5.4等,
其实还有一些更基础的readme没有提到,必备visual studio2015到2019中的一个版本等等,而且我是在win10中进行安装,还有一些问题需要解决。
接下来开始我的安装之路:
因为cuda10.2的安装需要依赖visual studio,而作者源码因为sh问题限定版本为vs2015-2019中的一个版本,这里我就安装vs2017,大家可以自行查找安装方法,请注意这个c++桌面开发是必选的,右侧可选用默认即可。
如果你的版本不是需求的版本(比如我刚开始的版本是vs2022),则要首先卸载CUDA和cudnn,如果用geek卸载会卸载的更干净一些,大家可以自行搜索geek这个巨好用的卸载工具。
这里推荐一个b站大佬的卸载和安装cuda和cudnn教程。
【【CUDA安装/多CUDA兼容】Windows深度学习环境配置】 【CUDA安装/多CUDA兼容】Windows深度学习环境配置_哔哩哔哩_bilibili
再卸载vs2022,这个大家也自行搜索卸载方法吧。
然后再安装vs2017,这样也可以省一点空间。
由于我自己常用pycharm,所以vs我只安装了c++的桌面开发,这样安装cuda就不会报错了。
这里推荐一个b站大佬的卸载和安装cuda和cudnn教程,只要按照他的来就没问题了,大家可以到第一个CUDA和CUDNN装好就结束视频。
【【CUDA安装/多CUDA兼容】Windows深度学习环境配置】 【CUDA安装/多CUDA兼容】Windows深度学习环境配置_哔哩哔哩_bilibili
安装git是为了跑后面的sh,make和bash,建议大家学习廖雪峰老师的git安装教程,简答且高效,链接如下:
安装Git - 廖雪峰的官方网站
我相信别的大家没装很正常,这个大家应该都装了…大家百度安装教程吧。
接下来我将一行注释一行代码的写,大家在anaconda的终端复制代码进去就行:
在anaconda3中建立新的虚拟环境
conda create -n owdetr python=3.7 pip
进入虚拟环境
conda activate owdetr
安装必要包
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch
这个要求要先进入
requirements.txt
的位置,大家
cd
到对应位置(如果是默认下载,则是进入
OW-DETR-main
文件夹)即可,大家不会的可以百度命令行
cd
怎么用,怎么换盘等。
pip install -r requirements.txt
跳出来问你yes/no的输入y再输入回车即可
安装完毕之后,需要打开git bash,如图,再一次输入到你放OW-DETR-main的文件夹,再输入
cd models
再输入
cd ops
进入ops文件夹
在这里,输入:
source activate owdetr
激活虚拟环境(我这里卡了1天,笑死了直接用base没进入虚拟环境,我说怎么一直安装失败)
再输入
sh ./make.sh
找到最上面,看看有没有
UserWarning: Error checking compiler version for cl: [WinError 2] ϵͳ▒Ҳ▒▒▒ָ▒▒▒▒▒ļ▒▒▒
warnings.warn(f'Error checking compiler version for {compiler}: {error}')
有的话就说明cl.exe没弄好,参照这篇博客调整
UserWarning: Error checking compiler version for cl: [WinError 2] 系统找不到指定的文件。_程序小K的博客-CSDN博客
这时应该向上检查,我在顶端部分看到
E:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include\crt/host_config.h(158): fatal error C1189: #error: -- unsupported Microsoft Visual Studio version! Only the versions between 2015 and 2019 (inclusive) are supported!
所以把vs重新装了2017,之后sh成功。
python test.py
运行成功则
python test.py
发现报错
ModuleNotFoundError: No module named 'MultiScaleDeformableAttention'
这不是没有包!而是没有sh ./make.sh失败
这里找到一篇教程
Windows配置MultiScaleDeformableAttention记录_仰夏ysummer的博客-CSDN博客
是cl.exe问题
大家可以参考该教程(其中重新安装vs可以不用做了,问题不在这里),配置环境变量(可以不用像教程一样新建,用原来的PATH在里面新建也可以),一定要记得一路确定回去,不然修改不会保存!
之后再继续按照教程里的走,记得用自己的安装位置,不要照抄!那些路径里面,如果数字部分不一样,请以自己的为准!
在test.py的channels中减少一下就行,测试嘛~