【SDXL】エロゲCGの生成AIを作る
WAI-NSFW-illustrious-SDXLをベースに、エロゲCG特化のSDXLモデルを作成中 学習方法は、100万枚くらいのエロゲCGにタグ+キャプションを付与し、SDXLのUNetを対象にフル微調整で学習させている WAIやIllustrious系はDanbooru由来の学習が多いと思われるけど、DanbooruにもエロゲCGはある程度入ってるのでそれっぽい出力はできる ただし完全に学習しきれてはいない感じで、特にエロシーンはパターンが少なく弱い印象 そこで今回のモデルでは、その辺を強化するのが目的 以下は同じプロンプトを使って生成した比較 自作モデルはまだ学習途中で、step 3000時点のチェックポイントから生成したもの prompt:1girl, breasts, nipples, long hair, large breasts, blush, 1boy, questionable, nude, smile, purple eyes, black hair, hairband, hair ribbon, ribbon, hetero, faceless, breast hold WAI: https://i.imgur.com/x4FxLfP.png 自作モデル: https://i.imgur.com/3sZS09p.png ちなみにまだSDXLを知らないという人は以下の自作WEBアプリで試すといい https://aiart.lovedeep.org 自宅PCを鯖にしてるのでいつでもリンク切れになる可能性あり 使ってるEmberVeilMixというモデルはWAIをベースにしたマージモデルだけどほぼWAIと変わらないと考えていい エロゲCGなんかを参考にしたら古臭くて見てられない絵になりそう そうかな? 少なくともソシャゲや同人ゲーと比べれば、まだマシな方だと思うけどね もちろん自分も有名どころ(ゆずソフトとか)くらいしか詳しくは知らないから、全体を語れるわけじゃない 今回学習に使ってるCGも、その辺のブランドの作品が中心になってるよ ちなみに学習にはVRAMが80GBくらい必要になるので、ローカル環境ではまず無理。 そのためクラウドGPUのA100 SXMをレンタルしてる。 費用は1時間あたり0.7ドルで、相場的には安いほうなんだけど、さすがに長時間は回せない。 とりあえず5万ステップくらいまで学習したら中止する予定。 本気度合いが凄いな 完成したらcivitaiに置いてくれよ そうだな どこまで進められるかはまだわからないけど 形になればCivitaiに置くかも 学習の進捗が分かるように、400ステップごとにプレビュー画像を生成している。 それをWebページ上で一覧できるようにしたので、下記からアクセスすれば現在のステップや生成画像のクオリティを確認できると思う http://198.13.48.172/resource/preview/viewer.html 今のところはベースモデルより劣っているのが見て取れるね このまま続けてもあまり良い結果になりそうにないので、パラメータを変えてやり直す予定 現在の設定: snr_gamma = 4 batch_size = 64 learning_rate = 1e-5 変更予定: snr_gamma = 1 batch_size = 32 learning_rate = 1e-6 結局3日間の学習は無駄になってしまった。 クラウドGPUの利用料として、だいたい1万円くらい消えた計算になる >>12 そうですね 特にWAIなどは絵師の絵柄も学習しているのでartistの名前を指定するとその絵師が描いたような絵も作れる 無料で利用できるアプリを作ったのでよかったら試してみてください https://aiart.lovedeep.org https://i.imgur.com/Z9eslpd.jpeg https://i.imgur.com/ejuTGzv.jpeg read.cgi ver 08.1h [pink] - 2023/09/20 Walang Kapalit ★ | uplift ★ 5ちゃんねる