mmrotate安装使用

在autoDL上边租一个GPU用来测试。
主要调试的代码是mmrotate里边的ReDet、Oriented R-CNN、S2ANet。我选这三个模型是因为这三个是截止2022年4月4日,HRSC2016数据集上的SOTA,我毕设在做目标检测,拿来凑论文材料。

0.租用GPU

我从autodl租的RTX 2080Ti,选配
PyTorch 1.7.0
Python 3.8
Cuda 11.0

1.安装mmrotate

我是按照官方的流程来的,网站在这里,可以参考
https://mmrotate.readthedocs.io/en/latest/install.html

1.1 conda创建openmmlab

conda create -n openmmlab python=3.7 -y
source activate openmmlab

1.2 配torchvision

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch

1.3安装 MMRotate依赖

他推荐用min安装mmrotate,所以要先pip安装openmim

pip install openmim
mim install mmrotate

1.4 git clone mmrotate

git clone https://github.com/open-mmlab/mmrotate.git
cd mmrotate
pip install -r requirements/build.txt
pip install -v -e .

安装完成了测试的时候发现mmrotate并不包含oriented r cnn关于hrsc2016的模型、结果等,只有dota的,崩溃啊,吃饭了。

最后的报错汇总

1.docker服务

github搜索mmrotate的时候,第二个市docker for mmrotate,我之前听说这玩意很好,结果ubuntu安装docker卡住好久,专业版docker还被美国制裁,其实我就是不想搞了,还是直接装mmrotate吧,

你可能感兴趣的:(复现论文,python)