旋转框目标检测mmrotate v0.3.1 环境搭建（一）

1、创建和激活虚拟环境

conda create -n mmrotate python=3.8 -y 
source activate mmrotate

2、Make sure your CUDA runtime api version ≤ CUDA driver version. (for example 10.1 ≤ 10.2)

nvcc -V  #查看CUDA运行版本
nvidia-smi  #查看CUDA驱动版本

3、安装pytorch

pip3 install torch==1.8.2+cu102 torchvision==0.9.2+cu102 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html  -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

#验证是否安装成功

>>> import torchvision
>>> import torch
>>> import.__version__
  File "<stdin>", line 1
    import.__version__
          ^
SyntaxError: invalid syntax
>>> torch.__version__
'1.8.2+cu102'

4、安装mmcv-full

pip install mmcv-full==1.3.17 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com #（镜像加速安装）

pip install mmcv-full==1.4.5 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com #（镜像加速安装）

assert (mmcv_version >= digit_version(mmcv_minimum_version)
AssertionError: MMCV==1.4.5 is used but incompatible. Please install mmcv>=1.2.4, <=1.4.0.

pip install mmcv-full==1.4.0 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
#验证是否安装成功
import mmcv

5、MMDetection安装

# 你可以使用下面的命令简单地安装mmdetection:
pip install mmdet
pip install mmdet==2.25.0  -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

#验证安装成功
>>> mmdet.__version__
'2.25.0'

6、MMRotate安装

git clone https://github.com/open-mmlab/mmrotate.git
cd mmrotate
pip install -r requirements/build.txt
pip install -v -e .  # or "python setup.py develop"

或者
pip install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com 
python setup.py develop

a.当指定-e或develop时，MMRotate安装在dev模式下，对代码进行的任何本地修改都将生效，无需重新安装。

7、另一个选项:Docker Image安装

我们提供一个Dockerfile来构建一个映像。确保您使用的docker版本添加链接描述大于=19.03。

# build an image with PyTorch 1.6, CUDA 10.1
docker build -t mmrotate docker/

Run it with

docker run --gpus all --shm-size=8g -it -v {DATA_DIR}:/mmrotate/data mmrotate

8、环境验证

为了验证MMRotate是否正确安装，我们可以运行演示代码并推断出演示图像。

1、我们提供了一个演示脚本来测试单个图像

python demo/image_demo.py \
    ${IMG_ROOT} \
    ${CONFIG_FILE} \
    ${CHECKPOINT_FILE}

#案例
python demo/image_demo.py demo/demo.jpg work_dirs/oriented_rcnn_r50_fpn_1x_dota_v3/oriented_rcnn_r50_fpn_1x_dota_v3.py work_dirs/oriented_rcnn_r50_fpn_1x_dota_v3/epoch_12.pth

#我的实践
python demo/image_demo.py demo/demo.jpg configs/oriented_rcnn/oriented_rcnn_r50_fpn_1x_dota_le90.py checkpoints/Oriented_RCNN/oriented_rcnn_r50_fpn_1x_dota_le90.pth

补充测试结果图，验证环境搭建成功

9、安装其他库

1、安装WBF库

pip install ensemble-boxes   -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com