前言
官方代码:
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
官方的给定环境条件: