ubuntu环境
- Ubuntu18.04
- CUDA 11.3
- conda3
记录一下mmrotate的环境配置,如果日后需要查阅,会方便一点。毕竟现在各个版本的代码太过繁杂,仅作为自己环境配置的记录!!!
建立新的conda环境
conda create -n mmrotate python=3.8 -y
source activate mmrotate
安装pytorch
按照官网所说的要求配置环境,本机安装的是cuda11.3,因此安装pytorch1.11.0版本
Python 3.7+, CUDA 9.2+, and PyTorch 1.8+
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
安装open-mmlab支撑库
使用 MIM 安装 MMCV 和 MMDetection.
pip install -U openmim
mim install mmcv-full
mim install mmdet\<3.0.0
不能下载最新版的mmdet,后续安装时回报错!!!!(ps:之前不想再配环境了,用的之前配好的最新版的mmdet。然后他就报错了。。。。。。)
下载mmrotate并安装
git clone https://github.com/open-mmlab/mmrotate.git
cd mmrotate
pip install -v -e .
# "-v" means verbose, or more output
# "-e" means installing a project in editable mode,
# thus any local modifications made to the code will take effect without reinstallation.
安装好验证一下~
mim download mmrotate --config oriented_rcnn_r50_fpn_1x_dota_le90 --dest .
python demo/image_demo.py demo/demo.jpg oriented_rcnn_r50_fpn_1x_dota_le90.py oriented_rcnn_r50_fpn_1x_dota_le90-6d2b2ce0.pth --out-file result.jpg
报错
AttributeError: 'ConfigDict' object has no attribute 'model'
进行 mmcv 与 mmdet 的版本更换,最后成功运行的版本为
MMCV:1.6.2
MMDet:2.26.0
更换版本再次运行,报错没有解决。(然后就开始了疯狂的更换环境,看源代码。。。。)
最后发现是下载 “ oriented_rcnn_r50_fpn_1x_dota_le90.py“ 和 "oriented_rcnn_r50_fpn_1x_dota_le90-6d2b2ce0.pth"文件的问题。由于网络问题,下载失败但是文件已经被创建,但是为0字节。(缩略图状态下谁会注意文件大小呀。。。。要不是脑子抽了,看了详细信息,估计还在程序版本上纠结呢。。。。)
重新下载后,就成功运行了~
放张成功运行的图片记录一下 ^ _ ^