DN-Detr环境配置

..Severus

已于 2022-07-24 16:24:08 修改

阅读量836

点赞数 1

分类专栏：深度学习文章标签： python pytorch 深度学习

于 2022-07-22 20:09:38 首次发布

本文链接：https://blog.csdn.net/qq_51123264/article/details/125938761

版权

4 篇文章 2 订阅

订阅专栏

官方代码：

https://github.com/IDEA-opensource/DN-DETR

官方论文：

https://arxiv.org/abs/2203.01305

一、创建一个新的环境：

conda create -n dn_detr python=3.8

激活：
二、激活：

conda activate dn_detr

三、下载pytorch的各种库：

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

这里注意一下，要检测下是否下下载的pytorch版本是不是GPU版本的，必须满足GPU版本，不然使用不了
在这里插入图片描述

四、然后cd进入主文件
在这里插入图片描述

安装requirements.txt文件：
官方的requirements.txt文件我改了下，panopticapi和pycocotools 我单独下载了不然下不下来，用我发的这个文件就行了。
改成我这样：
在这里插入图片描述

五、执行下面三条命令：

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/

pip install panopticapi -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

pip3 install pycocotools -i https://pypi.tuna.tsinghua.edu.cn/simple/ --trusted-host pypi.tuna.tsinghua.edu.cn

六、又进入…\DN-DETR-main\models\dn_dab_deformable_detr\ops文件下面
在这里插入图片描述

python setup.py  build install

python test.py

官方的给定环境条件：
在这里插入图片描述

关注