概要
W-Flowは、「Wasserstein Gradient Flowsを介したワンステップ生成モデリング」の公式PyTorchコードベースです。
このツールは、単一の順伝播でシンプルな参照分布からデータ分布へ画像を生成できる点が特徴です。
特にImageNet 256×256のような複雑なデータセットにおいて、1回のニューラルネットワーク評価で高品質な画像を効率的に生成し、優れたFID性能を達成します。
マルチステップモデルからの蒸留ではなく、ゼロから学習するモデルであり、Wasserstein勾配フローのダイナミクスに基づいて最適輸送更新を行います。
主に、高速かつ高品質な画像生成技術に関心のある研究者や開発者、特にPyTorch環境で生成モデルの実験や開発を行いたいユーザーを対象としています。
互換性・特徴
- PyTorch
- Python
- 画像生成
- 高速生成
- Colab対応
- Hugging Face対応
基本情報
| ライセンス | MIT |
| Stars | 53 |
| Forks | 4 |
| カテゴリ | 画像生成 |
| アクティビティ | low |