perfectgf/lora-dataset-studio

解決する課題

LoRA Dataset Studio は、LoRA トレーニング用の画像データセットを作成するライフサイクル全体に対して、統合されたブラウザベースのインターフェースを提供します。画像の収集、クリーニング、キャプション付け、トレーニングのために複数の異なるツールを行き来する必要がなくなり、AI モデルに特定のキャラクター、コンセプト、スタイルを教えるプロセスを効率化します。

仕組み

このソフトウェアは、複数の専門的な AI ツールを統合する中央ハブとして機能します。ユーザーのマシン上でローカルに実行され、実際のトレーニングプロセスには ai-toolkit を、ローカルでの画像生成とチェックポイントテストには ComfyUI を利用します。ワークフローをいくつかの段階に整理します。

• データセット管理： 画像のインポート、バンクへの整理、クロップ、ミラーリング、品質スコアリングによるキュレーション。
• キャプション付け： ローカルビジョンモデルまたは JoyCaption を使用して、画像の説明文やタグを生成。
• トレーニング： さまざまなモデルファミリー（SDXL、FLUX.1、Qwen-Image 2.1 など）向けのガイド付きレシピを提供し、トレーニングキューを管理。
• レビュー： 固定シードとプロンプトバッチを使用して異なるチェックポイントを比較し、トレーニングされた LoRA の品質を評価。

対象ユーザー

複雑なコマンドラインツールや手動のファイル整理を管理することなく、画像生成モデル用の高品質な LoRA を作成したい AI アーティストや開発者向けに設計されています。

ハイライト

• エンドツーエンドのワークフロー： 画像バンクのキュレーションやテキスト検出から、トレーニング、最終テストまでをカバー。
• 拡張可能なプラグインシステム： クラウドトレーニング（vast.ai 経由）、ウェブスクレイピング、Civitai や Hugging Face への公開用のオプションプラグインを提供。
• マルチモデルサポート： FLUX.1、SDXL、Krea 2 を含む幅広いトレーニングファミリーをサポート。
• 統合テスト： 系統とランキングツールを使用してチェックポイントと強度を比較する「スタジオ」を内蔵。