在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