morisato commited on
Commit
3ed4614
1 Parent(s): 19eebcc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +56 -1
README.md CHANGED
@@ -1,3 +1,58 @@
1
- ---
2
  license: unknown
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  license: unknown
2
+ language:
3
+ - ja
4
  ---
5
+
6
+ # 背景LoRA
7
+
8
+ 既存ののモデルではプロンプトで生成できない場所や風景を学習で生成できるようになったらいいなという実験です。
9
+
10
+ # karaokeroom.safetensors
11
+
12
+ [<img width="480" src="https://i.imgur.com/hclI0vj.jpg">](https://i.imgur.com/hclI0vj.jpg)
13
+ [<img width="480" src="https://i.imgur.com/8H3c7eE.jpg">](https://i.imgur.com/8H3c7eE.jpg)
14
+
15
+ カラオケ屋さんの部屋の雰囲気を学習したLoRA
16
+ Loraを読み込ませて、プロンプトに **karaokeroom** と記述してください。
17
+ プロンプトに、1girl, karaoke, microphone, 等とあわせて記述していただくとカラオケを歌ってる感じの絵ができます。
18
+
19
+ ※当LoRAを適用すると人物の描画や画風に影響が生じるようです。LoRAを適用するWeightを調整することで画風への影響を抑えられます。
20
+ 影響が気になった場合は <lora:karaokeroom:1>ではなく<lora:karaokeroom:0.6>といった感じで調整して使ってみてください。
21
+
22
+ ※karaokeroom, 1girl, karaoke, 等のプロンプトを書いても、部屋の風景のみで人物がうまく描画されないことがあります。
23
+ ガチャ要素があるのと、モデルによってはうまく働かない場合があるようです。その場合は根気よく何枚か生成してみるか、違うモデルを使ってみてください。
24
+
25
+ ## この実験をやってみた動機
26
+ たとえばプロンプトに shibuya,city, と書くと渋谷っぽい風景の絵が描いてくれます。これはモデルが「渋谷」という概念を知ってるという事だと思います。
27
+ しかし、 Nishinomiya と書いても西宮っぽい風景の絵を描いてくれません。これはモデルは「西宮」という概念を知らないという事だと思います。
28
+
29
+ 最近、LoRAという手法でスペックが低いパソコン(GPU)でも追加学習が出来る方法が普及してきました。既存のモデルでは描けないキャラクターや衣装等を学習させている方がたくさんいらっしゃいます。
30
+ そこで自分は、風景の写真を何枚か学習させれば、その「場所」の概念を学習してくれるのではないかと考えました。
31
+
32
+ カラオケ店の部屋の写真を20枚程用意して、WD14-taggerでタグ付けを行いました。出来たtxtファイルの全ての先頭の位置に karaokeroom, という単語を追加しました。
33
+ 学習前提のモデルは karaokeroom という概念を知らないので、この学習によって karaokeroom という新しい概念を獲得してくれると想定しました。
34
+
35
+ 実際に学習を実行して出来上がった当LoRAを読み込ませると上記の karaokeroom というプロンプトでカラオケ店の部屋っぽい絵が生成できます。
36
+ 場所の概念を学習させる実験は成功ではないでしょうか? 
37
+   
38
+ ## 問題点、今後の課題
39
+ カラオケ店の風景は再現できるようになりました。が、当LoRAを適用してカラオケを歌う女の子の絵を生成すると、人物の描画や画風に影響が生じる場合があります。
40
+ これはおそらく場所の概念だけではなく、素材写真の画風等も学習してしまったものだと思います。
41
+ 現状はLoraを適用するWeightを下げることで影響を軽減できますが、学習方法やLoRAの適用の仕方で影響を軽減することが出来ないか?と考えています。
42
+
43
+ ・U-net層でWeight調整することで影響を押さえられる?
44
+ 実は僕も全然よく分かってないのですが(!)
45
+ 階層マージ(Marge Block Weighted)で多くの方が様々なモデルマージに挑戦した結果、絵を生成するU-netの各レイヤー層を調節することで描画に様々な調節ができる(らしい?)ことがわかってきました。
46
+ 例えば「INの上層はリアル調、INの下層がanime調を担当しているのではないか?」、「M_00は全体にキャラクターや服装・背景等に大きな影響が出る」、「OUT上層は、主題以外の表現 (例えば背景)に影響を及ぼしている」、「OUT04,OUT05,OUT06あたりはめっちゃ顔に影響ある」等色々な仮説・検証・推論があります。
47
+
48
+ もし、背景に大きく関与しているU-net層が分かれば、それ以外のU-net層への関与を抑えることで、既存モデルの人物描写と追加学習背景LoRAでうまく共存できないだろうか?  
49
+ …と考えて、U-net層別にWeightを調節できるScripts(sd-webui-lora-block-weight:https://github.com/hako-mikan/sd-webui-lora-block-weight)を使って色々な数値を調整したXY Plot画像等を作成してみましたが…、現状では「何もわからないことがわかった」「自分が無力だってことだけ」という感��です。
50
+
51
+ ・学習時のキャプション(タグ)の付け方などで、画風は学ばないようにできないか?
52
+ ・正則化画像を用意することで何かうまく学習の調整ができるのでは?
53
+
54
+ 等が考えられると思いますが、まだ試行錯誤や情報が足りず良い解決方法は得られていません。
55
+
56
+
57
+
58
+