复现 R3Det-pytorch RTX3080ti

在RTX3080ti主机上复现R3Det-pytorch版本

前言

代码链接：GitHub - SJTU-Thinklab-Det/r3det-pytorch: R3Det based on mmdet 2.19.0

论文链接：https://arxiv.org/abs/1908.05612

显卡：RTX3080ti

系统：Ubuntu20.04 

一、创建环境

conda create -n pytorch171 python=3.7 -y
conda activate pytorch171
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

我这里和之前OBBDetection一样继续用1.7.1的torch 

二、配置R3Det

下载代码文件：

git clone https://github.com/SJTU-Thinklab-Det/r3det-pytorch.git --recursive
cd r3det-pytorch/

 安装mmcv：

登录这个网址，改网址里的数字就可以找对应torch和cuda版本的mmcv，注意下载的是mmcv-full

pip install mmcv-full==1.4.0 -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html --no-cache-dir

安装r3det：

pip install mmdet==2.19.0
pip install -r requirements.txt
pip install -v -e .

最后下载成功显示：

这个过程会花费几分钟。

Successfull