English | 简体中文
MMRotate 是一款基于 PyTorch 的旋转框检测的开源工具箱,是 OpenMMLab 项目的成员之一。
主分支代码目前支持 PyTorch 1.6 以上的版本。
支持多种角度表示法
MMRotate 提供了三种主流的角度表示法以满足不同论文的配置。
模块化设计
MMRotate 将旋转框检测任务解耦成不同的模块组件,通过组合不同的模块组件,用户可以便捷地构建自定义的旋转框检测算法模型。
强大的基准模型与SOTA
MMRotate 提供了旋转框检测任务中最先进的算法和强大的基准模型.
最新的 0.3.3 版本已经在 2022.10.27 发布:
如果想了解更多版本更新细节和历史信息,请阅读更新日志。
MMRotate 依赖 PyTorch, MMCV 和 MMDetection,以下是安装的简要步骤。
更详细的安装指南请参考 安装文档。
conda create -n open-mmlab python=3.7 pytorch==1.7.0 cudatoolkit=10.1 torchvision -c pytorch -y
conda activate open-mmlab
pip install openmim
mim install mmcv-full
mim install mmdet
git clone https://github.com/open-mmlab/mmrotate.git
cd mmrotate
pip install -r requirements/build.txt
pip install -v -e .
请参考快速入门文档学习 MMRotate 的基本使用。
我们提供了 colab 教程,也为新手提供了完整的运行教程,其他教程如下
各个模型的结果和设置都可以在对应的 config(配置)目录下的 README.md 中查看。
整体的概况也可也在 模型库 页面中查看。
请参考 data_preparation.md 进行数据集准备。
请参考 FAQ 了解其他用户的常见问题。
我们非常欢迎用户对于 MMRotate 做出的任何贡献,可以参考 CONTRIBUTION.md 文件了解更多细节。
MMRotate 是一款由不同学校和公司共同贡献的开源项目。我们感谢所有为项目提供算法复现和新功能支持的贡献者,以及提供宝贵反馈的用户。
我们希望该工具箱和基准测试可以为社区提供灵活的代码工具,供用户复现现有算法并开发自己的新模型,从而不断为开源社区提供贡献。
如果你在研究中使用了本项目的代码或者性能基准,请参考如下 bibtex 引用 MMRotate。
@inproceedings{zhou2022mmrotate,
title = {MMRotate: A Rotated Object Detection Benchmark using PyTorch},
author = {Zhou, Yue and Yang, Xue and Zhang, Gefan and Wang, Jiabao and Liu, Yanyi and
Hou, Liping and Jiang, Xue and Liu, Xingzhao and Yan, Junchi and Lyu, Chengqi and
Zhang, Wenwei and Chen, Kai},
booktitle={Proceedings of the 30th ACM International Conference on Multimedia},
year={2022}
}
该项目采用 Apache 2.0 license 开源协议。
扫描下方的二维码可关注 OpenMMLab 团队的 知乎官方账号,加入 OpenMMLab 团队的 官方交流 QQ 群或联络 OpenMMLab 官方微信小助手
我们会在 OpenMMLab 社区为大家
干货满满 📘,等你来撩 💗,OpenMMLab 社区期待您的加入 👬
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》