• 安装MMRotate流程


    一、介绍

    MMRotate的github网站已经给出了非常详细的安装步骤
    所有的安装步骤已经在安装文档里,并且非常详细,
    本文主要记录一些踩坑的地方。

    二、流程

    1.配置conda环境

    假设你的Anaconda已经安装好了,并且之前自己创建过环境。使用 nvidia-smi 指令查询,我的CUDA版本是12.1。

    cmd打开本地终端(最好是在D盘下面打开,因为后续的mmrotate的文件会直接安装在当前目录)

    conda create --name openmmlab python=3.8 -y
    conda activate openmmlab
    
    • 1
    • 2
    conda install pytorch==1.8.0 torchvision==0.9.0 cudatoolkit=10.2 -c pytorch
    
    • 1

    2.使用 MIM 安装 MMCV 和 MMDetection
    非常建议还是使用MIM安装,否则的话还要去根据 PyTorch 版本及其 CUDA 版本手动指定下载指令后面的 find-url,比较麻烦)

    注意,github上的安装步骤其实是有一点点问题的,就是我按照如下:

    pip install -U openmim
    mim install mmcv-full
    mim install mmdet

    安装的话,最后运行mmrotate的时候会报错:

    `MMCV 与 MMDetection 的兼容问题: “ConvWS is already registered in conv layer”; “AssertionError: MMCV==xxx is used but incompatible. Please install mmcv>=xxx, <=1.8.0

    所以我又重新配置了环境,并使用:

    pip install -U openmim
    mim install mmcv-full==1.5.3
    mim install mmdet==2.25.1
    
    • 1
    • 2
    • 3

    进行安装。
    (具体版本对应如下所示,见此链接
    在这里插入图片描述3.安装 MMRotate
    我直接从源码安装

    git clone https://github.com/open-mmlab/mmrotate.git
    cd mmrotate
    pip install -v -e .
    
    • 1
    • 2
    • 3

    4.验证
    (1)首先下载配置文件和检查点文件

    mim download mmrotate --config oriented_rcnn_r50_fpn_1x_dota_le90 --dest .
    
    • 1

    (2)验证推理演示

    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
    
    • 1

    然后会在当前目录下看到一张名为 result.jpg 的新图片,其中旋转边界框绘制在汽车、公共汽车等目标上,就算成功啦
    在这里插入图片描述
    (3)注意:在这一步骤的时候,可能会报错:

      File "D:\Anaconda\envs\open_mmlab\lib\site-packages\PIL\Image.py", line 100, in <module>
        from . import _imaging as core
    ImportError: DLL load failed while importing _imaging: 找不到指定的模块。
    
    • 1
    • 2
    • 3

    这可能是因为你的Pillow版本不匹配,
    解决办法:先卸载Pillow再安装就行了

    pip uninstall Pillow
    pip install Pillow
    
    • 1
    • 2
  • 相关阅读:
    前端必掌握的Vue理论知识
    暑期JAVA学习(43.2)反射
    0015Java程序设计-springboot美食网站
    ubuntu20 ros使用usb相机
    C++基础知识(十五)--- deque 容器
    R3LIVE代码详解(四)
    驱动开发:内核特征码扫描PE代码段
    《HelloGitHub》第 74 期
    图的邻接表存储
    1196: 数星星(二)(结构体专题)
  • 原文地址:https://blog.csdn.net/qq_47104322/article/details/133073980