在autoDL上边租一个GPU用来测试。
主要调试的代码是mmrotate里边的ReDet、Oriented R-CNN、S2ANet。我选这三个模型是因为这三个是截止2022年4月4日,HRSC2016数据集上的SOTA,我毕设在做目标检测,拿来凑论文材料。
我从autodl租的RTX 2080Ti,选配
PyTorch 1.7.0
Python 3.8
Cuda 11.0
我是按照官方的流程来的,网站在这里,可以参考
https://mmrotate.readthedocs.io/en/latest/install.html
conda create -n openmmlab python=3.7 -y
source activate openmmlab
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch
他推荐用min安装mmrotate,所以要先pip安装openmim
pip install openmim
mim install 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的,崩溃啊,吃饭了。
github搜索mmrotate的时候,第二个市docker for mmrotate,我之前听说这玩意很好,结果ubuntu安装docker卡住好久,专业版docker还被美国制裁,其实我就是不想搞了,还是直接装mmrotate吧,