mmgenerationダウンロードmmgenerationソースコードのダウンロード

mmgeneration

パイソン

v0.7.3 Released

ダウンロード

英語| 简体中文

導入

MMGenerationは、特に現在のGANにとっては、生成モデルにとって強力なツールキットです。 PytorchとMMCVに基づいています。マスターブランチは、Pytorch 1.5+で動作します。

主な機能

高品質のトレーニングパフォーマンス：現在、無条件GAN、内部GAN、画像翻訳モデルに関するトレーニングをサポートしています。条件付きモデルのサポートはまもなく行われます。
強力なアプリケーションツールキット： GANに複数のアプリケーションを含む豊富なツールキットがユーザーに提供されます。 GAN補間、GAN予測、およびGANの操作は、フレームワークに統合されています。あなたのガンと遊ぶ時が来ました！（アプリケーションのチュートリアル）
生成モデル向けの効率的な分散トレーニング：生成モデルでの非常に動的なトレーニングのために、 MMDDPで動的モデルをトレーニングする新しい方法を採用します。（DDPのチュートリアル）
柔軟な組み合わせのための新しいモジュラー設計：複雑な損失モジュールの新しい設計が、モジュール間のリンクをカスタマイズするために提案されています。これにより、異なるモジュール間の柔軟な組み合わせを実現できます。（新しいモジュラーデザインのチュートリアル）

視覚化のトレーニング	ガン補間	GANプロジェクター	ガン操作

ハイライト

GANS（CVPR2021）の空間誘導バイアスとしての位置エンコーディングは、 MMGenerationで放出されています。 [config]、[プロジェクトページ]
条件付きガンは、私たちのツールキットでサポートされています。より多くの方法と事前に訓練されたウェイトがまもなく行われます。
StyleGan2の混合精度トレーニング（FP16）がサポートされています。異なる実装の比較を確認してください。

Changelog

V0.7.3は14/04/2023にリリースされました。詳細とリリース履歴については、changelog.mdを参照してください。

インストール

mmgenerationは、PytorchとMMCVに依存します。以下は、インストール用の簡単な手順です。

ステップ1。公式の指示に従ってpytorchをインストールします

 pip3 install torch torchvision

ステップ2。MIMでMMCVをインストールします。

 pip3 install openmim
mim install mmcv-full

ステップ3。ソースからmmgenerationをインストールします。

 git clone https://github.com/open-mmlab/mmgeneration.git
cd mmgeneration
pip3 install -e .

詳細な指示については、get_started.mdを参照してください。

はじめる

mmgenerationの基本的な使用については、get_started.mdを参照してください。 docs/en/Quick_run.mdは、クイックランのための完全なガイダンスを提供できます。その他の詳細やチュートリアルについては、ドキュメントをご覧ください。

ModelZoo

これらの方法は、フレームワークで慎重に研究およびサポートされています。

無条件のガン（クリックして崩壊）

✅DCGAN（ICLR'2016）
✅wgan-gp（nips'2017）
✅LSGAN（ICCV'2017）
✅GGAN（arxiv'2017）
✅PGGAN（ICLR'2018）
✅styleganv1（cvpr'2019）
✅styleganv2（cvpr'2020）
✅styleganv3（neurips'2021）
GANSでの位置エンコーディング（CVPR'2021）

条件付きガン（クリックして崩壊）

✅スナン（ICLR'2018）
✅プロジェクションガン（ICLR'2018）
✅サガン（ICML'2019）
biggan/biggan-deep（iclr'2019）

ガンのトリック（クリックして崩壊）

✅ada（neurips'2020）

image2imageの翻訳（クリックして崩壊します）

pix2pix（cvpr'2017）
✅Cyclegan（ICCV'2017）

内部学習（クリックして崩壊）

✅シンガン（ICCV'2019）

拡散確率モデルを除去する（クリックして崩壊する）

ddpmの改善（arxiv'2021）

貢献

私たちは、mmgenerationを改善するためのすべての貢献に感謝しています。貢献ガイドラインの詳細については、MMCVのContributing.mdを参照してください。

引用

このプロジェクトがあなたの研究で役立つと思う場合は、引用を検討してください。

 @misc { 2021mmgeneration ,
    title = { {MMGeneration}: OpenMMLab Generative Model Toolbox and Benchmark } ,
    author = { MMGeneration Contributors } ,
    howpublished = { url{https://github.com/open-mmlab/mmgeneration} } ,
    year = { 2021 }
}