Win10 训练OWDETR(OW-DETR)

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中进行安装,还有一些问题需要解决。

Win10 训练OWDETR(OW-DETR)_第1张图片

接下来开始我的安装之路:

  • 1.安装visual studio 2017

因为cuda10.2的安装需要依赖visual studio,而作者源码因为sh问题限定版本为vs2015-2019中的一个版本,这里我就安装vs2017,大家可以自行查找安装方法,请注意这个c++桌面开发是必选的,右侧可选用默认即可。

Win10 训练OWDETR(OW-DETR)_第2张图片

如果你的版本不是需求的版本(比如我刚开始的版本是vs2022),则要首先卸载CUDA和cudnn,如果用geek卸载会卸载的更干净一些,大家可以自行搜索geek这个巨好用的卸载工具。

这里推荐一个b站大佬的卸载和安装cuda和cudnn教程。

【【CUDA安装/多CUDA兼容】Windows深度学习环境配置】 【CUDA安装/多CUDA兼容】Windows深度学习环境配置_哔哩哔哩_bilibili

再卸载vs2022,这个大家也自行搜索卸载方法吧。

然后再安装vs2017,这样也可以省一点空间。

由于我自己常用pycharm,所以vs我只安装了c++的桌面开发,这样安装cuda就不会报错了。

  • 2.安装CUDA和CUDNN

这里推荐一个b站大佬的卸载和安装cuda和cudnn教程,只要按照他的来就没问题了,大家可以到第一个CUDA和CUDNN装好就结束视频。

【【CUDA安装/多CUDA兼容】Windows深度学习环境配置】 【CUDA安装/多CUDA兼容】Windows深度学习环境配置_哔哩哔哩_bilibili

  • 3.安装git

安装git是为了跑后面的sh,make和bash,建议大家学习廖雪峰老师的git安装教程,简答且高效,链接如下:

安装Git - 廖雪峰的官方网站

  • 4.安装anaconda3

我相信别的大家没装很正常,这个大家应该都装了…大家百度安装教程吧。

  • 5.按照readme安装教程行动

接下来我将一行注释一行代码的写,大家在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文件夹

Win10 训练OWDETR(OW-DETR)_第3张图片

在这里,输入:

source activate owdetr

激活虚拟环境(我这里卡了1天,笑死了直接用base没进入虚拟环境,我说怎么一直安装失败)

再输入

sh ./make.sh

这里就算最后显示finished也可能失败

找到最上面,看看有没有

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博客

最后可能出现失败,比如RuntimeError

这时应该向上检查,我在顶端部分看到

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成功。

安装成功

  • 6.按照教程进行测试

在终端运行(注意请先cd进入ops文件夹)

python test.py

运行成功则

Win10 训练OWDETR(OW-DETR)_第4张图片

如果ModuleNotFoundError: No module named 'MultiScaleDeformableAttention'

python test.py

发现报错

ModuleNotFoundError: No module named 'MultiScaleDeformableAttention'

这不是没有包!而是没有sh ./make.sh失败

这里找到一篇教程

Windows配置MultiScaleDeformableAttention记录_仰夏ysummer的博客-CSDN博客

是cl.exe问题

大家可以参考该教程(其中重新安装vs可以不用做了,问题不在这里),配置环境变量(可以不用像教程一样新建,用原来的PATH在里面新建也可以),一定要记得一路确定回去,不然修改不会保存!

之后再继续按照教程里的走,记得用自己的安装位置,不要照抄!那些路径里面,如果数字部分不一样,请以自己的为准!

如果GPU不够用

在test.py的channels中减少一下就行,测试嘛~

Win10 训练OWDETR(OW-DETR)_第5张图片

  • 7.之后我要写一个端到端的演示模型,等我好消息。或者哪个大佬告诉我文件中有没有对单个图片进行测试的程序。

你可能感兴趣的:(深度学习,人工智能,transformer,目标检测,视觉检测)