HAN&DAI

  • 首页
  • 遥感应用
  • GIS应用
  • 机器学习
  • 实用工具
  • 文章链接
  • 遥感数据集
HAN&DAI
遥感与地理信息技术交流社区
  1. 首页
  2. 遥感应用
  3. 正文

mmrotate旋转框目标检测框架部署

2024年4月1日 448点热度 1人点赞 0条评论

提出问题

当前,有众多的开源方法集合以及社区维护。为了满足旋转框目标检测需求,一些学者在pytorch深度学习框架基础上构建了mmrotate旋转框目标检测体系。借此机会,自己搭建了一下mmrotate的环境,加深学习和为后续做相关对比实验做一些准备工作。

解决方案

  1. 一台2080或者2080ti的机器,cuda 10.2
  2. 参考代码: https://github.com/open-mmlab/mmrotate/

环境部署步骤

  1. 首先要下载两个代码和相关库,链接分别为:mmrotate、mmdetection和mmcv。这三个库的版本配置可参考该链接:相关版本适配
  2. 部署步骤或命令
    (1)创建python 3.8版本环境

    conda create --name openmmlab python=3.8 -y

    (2) 构建pytorch环境

    conda install pytorch==1.8.0 torchvision==0.9.0 cudatoolkit=10.2 -c pytorch

    (3)安装openmim

    pip install openmim

    (4)安装mmcv-full

    #从第一大步骤中的mmcv链接中下载1.6.2版本的mmcv到本地,然后执行安装
    pip install mmcv_full-1.6.2-cp38-cp38-manylinux1_x86_64.whl

    (5)安装mmdet,直接执行pip install mmdet可能会安装高版本的mmdet,导致最终环境不适配,因此,我们从第一大步骤中的mmdetection链接中下载代码到本地,然后执行相关命令安装。

    cd mmdetection
    pip install -r requirements/build.txt
    pip install -v -e .

    (6)安装mmrotate

    cd mmrotate
    pip install -r requirements/build.txt
    pip install -v -e .

    (7)安装后即可根据相关例子进行学习。

备注:这里提供一份能够直接学习的mmrotate的代码,用于faster rcnn训练少量ssdd的sar数据集。
百度网盘,提取码在评论区留言后发出。

Post Views: 447

相关文章:

  1. DarkLabel2.4软件标注视频影像数据做目标检测/跟踪(数据预处理)
  2. 利用GEE下载指定区域Landsat8影像
  3. 大尺度遥感线性地物(道路)提取后细化(骨架提取)
  4. YOLOv5+DeepSORT实现基于检测的视频目标跟踪
本作品采用 知识共享署名 4.0 国际许可协议 进行许可
标签: pytorch 环境配置 目标检测 遥感影像应用
最后更新:2024年5月4日

HAN&DAI

RS和GIS研究兴趣者,永远在学习的路上!

打赏 点赞
< 上一篇
下一篇 >

文章评论

razz evil exclaim smile redface biggrin eek confused idea lol mad twisted rolleyes wink cool arrow neutral cry mrgreen drooling persevering
取消回复

文章目录
  • 提出问题
  • 解决方案
  • 环境部署步骤
浏览最多的文章
  • BUG:ImportError: /lib/x86_64-linux-gnu/libstdc++.so.6: version `GLIBCXX_3.4.29' not found (1,462)
  • BUG:“ModuleNotFoundError: No module named '_ext'”的解决方案 (1,229)
  • 利用GEE下载指定区域Landsat8影像 (1,175)
  • 利用arcgis制作深度学习标签数据(以二分类为例) (899)
  • 利用传统机器学习方法进行遥感影像分类-以随机森林(RF)为例 (807)

COPYRIGHT © 2025 HAN&DAI. ALL RIGHTS RESERVED. QQ交流群:821388027

Theme Kratos Made By Seaton Jiang