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

(OpenMMLab Rotated Object Detection Toolbox and Benchmark)

Created at: 2022-02-17 09:19:05
Language: Python
License: Apache-2.0

English |简体中文

序章

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

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

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

主な特長
  • 複数の角度表現をサポート

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

  • モジュール設計

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

  • 強力なベースラインと最新技術

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

新着情報

0.3.2は 2022 年 6 月 7 日にリリースされました。

  • 指向的なリポイントをサポート (CVPR'22) (#286)
  • ConvNeXt バックボーン (CVPR'22) をサポート (#343)

詳細とリリース履歴については、 changelog.mdを参照してください。

インストール

MMRotate はPyTorchMMCV、および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のページにあります。

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

データ準備

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 モデル展開フレームワーク。