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 ウェブサイト^HOTOpenMMLab プラットフォーム^{試してみよう}

英語|简体中文

紹介

MMRotate は、PyTorch をベースにした回転オブジェクト検出のためのオープンソースのツールボックスです。これはOpenMMLabプロジェクトの一部です。

マスターブランチは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を参照してください。コラボチュートリアルとその他のチュートリアルを提供します。

モデル動物園

結果とモデルは、各メソッドの config ディレクトリのREADME.mdにあります。概要はモデル動物園のページにあります。

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

[バツ]回転網膜ネット-OBB/HBB(ICCV'2017)
[バツ]回転速度を上げるRCNN-OBB(TPAMI'2017)
[バツ]回転したレプポイント-OBB(ICCV'2019)
[バツ]ローテーションFCOS(ICCV'2019)
[バツ]ロイトランスフォーマー(CVPR'2019)
[バツ]滑空頂点(TPAMI'2020)
[バツ]回転ATSS-OBB(CVPR'2020)
[バツ]CSL(ECCV'2020)
[バツ]R³Det(AAAI'2021)
[バツ]S²Aネット(TGRS'2021)
[バツ]リデット(CVPR'2021)
[バツ]バウンディングボックスを超えて(CVPR'2021)
[バツ]オリエンテッドR-CNN(ICCV'2021)
[バツ]GWD(ICML'2021)
[バツ]KLDオランダ航空(ニューロIPS'2021)
[バツ]サズム(AAAI'2022)
[バツ]オリエンテッドレプポイント(CVPR'2022)
[バツ]KFIoU(arXiv)
[バツ]G-Rep(乞うご期待)

データ準備

データを準備するには、data_preparation.mdを参照してください。

FAQ

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

貢献

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

承認

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 ライセンスの下でリリースされています。

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 fewshot learning toolbox and benchmark.
MMAction2: OpenMMLab の次世代アクション理解ツールボックスとベンチマーク。
MMTracking: OpenMMLab ビデオ認識ツールボックスとベンチマーク。
MMFlow: OpenMMLab オプティカルフローツールボックスとベンチマーク。
MMEditing: OpenMMLab の画像およびビデオ編集ツールボックス。
MMGeneration: OpenMMLab 画像およびビデオ生成モデルツールボックス。
MMDeploy: OpenMMLab モデル展開フレームワーク。