mmrotate安装使用

mmrotate安装使用,第1张

在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吧,

欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/langs/571058.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-09
下一篇 2022-04-09

发表评论

登录后才能评论

评论列表(0条)

保存