Contents

安装mmrotate

需要下载Visual studio 2019(卸载之前的2022,如果有安装的话)

  • 下载页面 https://learn.microsoft.com/en-us/visualstudio/releases/2019/history#release-dates-and-build-numbers

  • 选择 BuildTools 链接并下载,安装之后,选择可安装页面中的 社区版安装,并在更改中选择桌面c++开发安装

  • 配置cl path的环境变量,例如 C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30133\bin\Hostx86\x64

根据cuda版本,查找对应的pytorch: https://pytorch.ac.cn/get-started/previous-versions/

查找mmcv离线版,示例链接 https://download.openmmlab.com/mmcv/dist/cu118/torch2.0.0/index.html ,需替换 cu118 torch2.0.0 中的版本号,下载最下方的 mmcv_full 文件,cp代表python版本

安装mmrotate环境

1
2
3
4
5
conda create -n mmrotate_env_11.8 python=3.10 -y
conda activate mmrotate_env_11.8 
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia
pip install H:\Workspace\python\mmcv_full-1.7.2-cp310-cp310-win_amd64.whl
pip install -U openmim

下载mmrotate https://github.com/open-mmlab/mmrotate/archive/refs/heads/main.zip ,解压后文件夹改名为mmrotate

1
2
3
cd mmrotate
conda activate mmrotate_env_11.8 
pip install -v -e .

下载模型和配置

  • 可以到这个页面下载 https://github.com/open-mmlab/mmrotate/blob/main/configs/r3det/README.md
  • 也可以运行这个命令下载 mim download mmrotate --config r3det_r50_fpn_1x_dota_oc --dest .
  • 配置文件移到 mmrotate\configs\r3det\r3det_r50_fpn_1x_dota_oc.py
  • 模型文件移到 mmrotate\checkpoint\r3det_r50_fpn_1x_dota_oc-b1fb045c.pth

测试mmrotate,如果成功,mmrotate目录下会生成 result.jpg

1
python demo/image_demo.py demo/demo.jpg configs/r3det/r3det_r50_fpn_1x_dota_oc.py checkpoint/r3det_r50_fpn_1x_dota_oc-b1fb045c.pth --out-file result.jpg --device=cpu

参考

coffee