mmrotate - OpenMMLab 回転物体検出ツールボックスとベンチマーク

(OpenMMLab Rotated Object Detection Toolbox and Benchmark)

Created at: 2022-02-17 09:19:05

Language: Python

URL: https://github.com/open-mmlab/mmrotate

License: Apache-2.0

OpenMMLab の Web サイト ^ホットOpenMMLab プラットフォーム^{TRY IT OUT}

English |简体中文

序章

MMRotate は、PyTorch に基づく回転オブジェクト検出用のオープンソースツールボックスです。これはOpenMMLab プロジェクトの一部です。

master ブランチはPyTorch 1.6+で動作します。

https://user-images.githubusercontent.com/10410257/154433305-416d129b-60c8-44c7-9ebb-5ba106d3e9d5.MP4

主な特長

複数の角度表現をサポート

MMRotate は、さまざまな用紙設定に対応するために、3 つの主流の角度表現を提供します。
モジュール設計

回転オブジェクト検出フレームワークをさまざまなコンポーネントに分解します。これにより、さまざまなモジュールを組み合わせて新しいモデルを簡単かつ柔軟に構築できます。
強力なベースラインと最新技術

このツールボックスは、回転物体検出における強力なベースラインと最先端の方法を提供します。

インストール

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の基本的な使用方法については、get_started.mdを参照してください。colab チュートリアル、およびその他のチュートリアルを提供します。

模型動物園

結果とモデルは、各メソッドの config ディレクトリのREADME.mdで入手できます。概要は、Model Zooのページにあります。

サポートされているアルゴリズム:

[x]回転した RetinaNet-OBB/HBB (ICCV'2017)
[x] Rotated FasterRCNN-OBB (TPAMI'2017)
[x]ローテーションされた RepPoints-OBB (ICCV'2019)
[x]ローテーションFCOS (ICCV'2019)
[x] RoI Transformer (CVPR'2019)
[x]グライディングバーテックス(TPAMI'2020)
[x]回転 ATSS-OBB (CVPR'2020)
[x] CSL (ECCV'2020)
[x] R ³ Det (AAAI'2021)
[x] S ² A-Net (TGRS'2021)
[x] ReDet (CVPR'2021)
[x]境界ボックスを超えて(CVPR'2021)
[x]指向性 R-CNN (ICCV'2021)
[x] GWD (ICML'2021)
[x] KLD (NeurIPS'2021)
[x] SASM (AAAI'2022)
[x] RepPoint 指向(CVPR'2022)
[x] KFIoU (arXiv)
[x] G-Rep (お楽しみに)

データ準備

data_preparation.mdを参照してデータを準備してください。

よくある質問

よくある質問については、 FAQを参照してください。

貢献する

MMRotate を改善するためのすべての貢献に感謝します。寄稿のガイドラインについては、 CONTRIBUTING.mdを参照してください。

了承

MMRotate は、さまざまな大学や企業の研究者やエンジニアによって貢献されたオープンソースプロジェクトです。メソッドを実装したり、新しい機能を追加したりするすべての貢献者、および貴重なフィードバックを提供するユーザーに感謝します。ツールボックスとベンチマークが、既存の方法を再実装し、独自の新しい方法を開発するための柔軟なツールキットを提供することにより、成長する研究コミュニティに役立つことを願っています.

引用

このツールボックスまたはベンチマークを研究で使用する場合は、このプロジェクトを引用してください。

@article{mmrotate2022,
  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},
  journal= {arXiv preprint arXiv:2204.13317},
  year={2022}
}

ライセンス

このプロジェクトは、Apache 2.0 ライセンスの下でリリースされています。

OpenMMLab のプロジェクト

MMCV : コンピュータビジョン用の OpenMMLab 基本ライブラリ。
MIM : MIM は OpenMMLab パッケージをインストールします。
MMClassification : OpenMMLab 画像分類ツールボックスおよびベンチマーク。
MMDetection : OpenMMLab 検出ツールボックスおよびベンチマーク。
MMDetection3D : 一般的な 3D オブジェクト検出のための OpenMMLab の次世代プラットフォーム。
MMRotate : OpenMMLab 回転オブジェクト検出ツールボックスおよびベンチマーク。
MMSegmentation : OpenMMLab セマンティックセグメンテーションツールボックスおよびベンチマーク。
MMOCR : OpenMMLab テキスト検出、認識、および理解ツールボックス。
MMPose : OpenMMLab 姿勢推定ツールボックスおよびベンチマーク。
MMHuman3D : OpenMMLab 3D ヒューマンパラメトリックモデルツールボックスおよびベンチマーク。
MMSelfSup : OpenMMLab 自己教師あり学習ツールボックスおよびベンチマーク。
MMRazor : OpenMMLab モデル圧縮ツールボックスおよびベンチマーク。
MMFewShot : OpenMMLab の少数ショット学習ツールボックスとベンチマーク。
MMAction2 : OpenMMLab の次世代アクション理解ツールボックスおよびベンチマーク。
MMTracking : OpenMMLab ビデオ認識ツールボックスおよびベンチマーク。
MMFlow : OpenMMLab オプティカルフローツールボックスおよびベンチマーク。
MMEditing : OpenMMLab 画像およびビデオ編集ツールボックス。
MMGeneration : OpenMMLab 画像およびビデオ生成モデルツールボックス。
MMDeploy : OpenMMLab モデル展開フレームワーク。