概要
このツールは、拡散モデルのトレーニング（SDXLのようなマルチARバッチ）において、アスペクト比の異なる画像を効率的にバッチ処理するためのライブラリです。
画像を歪ませたり破棄したりすることなく、類似のアスペクト比を持つ画像をバケットにまとめ、PyTorchネイティブかつDDP対応のサンプラーを提供します。
SDXLなどのプリセットやHuggingFaceデータセットとの連携、データ分析用CLIを備え、多様なアスペクト比の画像データセットを扱う開発者や研究者に適しています。
互換性・特徴
- Python
- PyTorch
- CLI
- HuggingFace
- 拡散モデル
- データ前処理
基本情報
| ライセンス | MIT |
| Stars | 1 |
| カテゴリ | 画像生成 |
| アクティビティ | low |