Skip to content

Latest commit

 

History

History
386 lines (288 loc) · 24.1 KB

config_README-ja.md

File metadata and controls

386 lines (288 loc) · 24.1 KB

--dataset_config で渡すことができる設定ファイルに関する説明です。

概要

設定ファイルを渡すことにより、ユーザが細かい設定を行えるようにします。

  • 複数のデータセットが設定可能になります
    • 例えば resolution をデータセットごとに設定して、それらを混合して学習できます。
    • DreamBooth の手法と fine tuning の手法の両方に対応している学習方法では、DreamBooth 方式と fine tuning 方式のデータセットを混合することが可能です。
  • サブセットごとに設定を変更することが可能になります
    • データセットを画像ディレクトリ別またはメタデータ別に分割したものがサブセットです。いくつかのサブセットが集まってデータセットを構成します。
    • keep_tokensflip_aug 等のオプションはサブセットごとに設定可能です。一方、resolutionbatch_size といったオプションはデータセットごとに設定可能で、同じデータセットに属するサブセットでは値が共通になります。詳しくは後述します。

設定ファイルの形式は JSON か TOML を利用できます。記述のしやすさを考えると TOML を利用するのがオススメです。以下、TOML の利用を前提に説明します。

TOML で記述した設定ファイルの例です。

[general]
shuffle_caption = true
caption_extension = '.txt'
keep_tokens = 1

# これは DreamBooth 方式のデータセット
[[datasets]]
resolution = 512
batch_size = 4
keep_tokens = 2

  [[datasets.subsets]]
  image_dir = 'C:\hoge'
  class_tokens = 'hoge girl'
  # このサブセットは keep_tokens = 2 (所属する datasets の値が使われる)

  [[datasets.subsets]]
  image_dir = 'C:\fuga'
  class_tokens = 'fuga boy'
  keep_tokens = 3

  [[datasets.subsets]]
  is_reg = true
  image_dir = 'C:\reg'
  class_tokens = 'human'
  keep_tokens = 1

# これは fine tuning 方式のデータセット
[[datasets]]
resolution = [768, 768]
batch_size = 2

  [[datasets.subsets]]
  image_dir = 'C:\piyo'
  metadata_file = 'C:\piyo\piyo_md.json'
  # このサブセットは keep_tokens = 1 (general の値が使われる)

この例では、3 つのディレクトリを DreamBooth 方式のデータセットとして 512x512 (batch size 4) で学習させ、1 つのディレクトリを fine tuning 方式のデータセットとして 768x768 (batch size 2) で学習させることになります。

データセット・サブセットに関する設定

データセット・サブセットに関する設定は、登録可能な箇所がいくつかに分かれています。

  • [general]
    • 全データセットまたは全サブセットに適用されるオプションを指定する箇所です。
    • データセットごとの設定及びサブセットごとの設定に同名のオプションが存在していた場合には、データセット・サブセットごとの設定が優先されます。
  • [[datasets]]
    • datasets はデータセットに関する設定の登録箇所になります。各データセットに個別に適用されるオプションを指定する箇所です。
    • サブセットごとの設定が存在していた場合には、サブセットごとの設定が優先されます。
  • [[datasets.subsets]]
    • datasets.subsets はサブセットに関する設定の登録箇所になります。各サブセットに個別に適用されるオプションを指定する箇所です。

先程の例における、画像ディレクトリと登録箇所の対応に関するイメージ図です。

C:\
├─ hoge  ->  [[datasets.subsets]] No.1  ┐                        ┐
├─ fuga  ->  [[datasets.subsets]] No.2  |->  [[datasets]] No.1   |->  [general]
├─ reg   ->  [[datasets.subsets]] No.3  ┘                        |
└─ piyo  ->  [[datasets.subsets]] No.4  -->  [[datasets]] No.2   ┘

画像ディレクトリがそれぞれ1つの [[datasets.subsets]] に対応しています。そして [[datasets.subsets]] が1つ以上組み合わさって1つの [[datasets]] を構成します。[general] には全ての [[datasets]], [[datasets.subsets]] が属します。

登録箇所ごとに指定可能なオプションは異なりますが、同名のオプションが指定された場合は下位の登録箇所にある値が優先されます。先程の例の keep_tokens オプションの扱われ方を確認してもらうと理解しやすいかと思います。

加えて、学習方法が対応している手法によっても指定可能なオプションが変化します。

  • DreamBooth 方式専用のオプション
  • fine tuning 方式専用のオプション
  • caption dropout の手法が使える場合のオプション

DreamBooth の手法と fine tuning の手法の両方とも利用可能な学習方法では、両者を併用することができます。 併用する際の注意点として、DreamBooth 方式なのか fine tuning 方式なのかはデータセット単位で判別を行っているため、同じデータセット中に DreamBooth 方式のサブセットと fine tuning 方式のサブセットを混在させることはできません。 つまり、これらを併用したい場合には異なる方式のサブセットが異なるデータセットに所属するように設定する必要があります。

プログラムの挙動としては、後述する metadata_file オプションが存在していたら fine tuning 方式のサブセットだと判断します。 そのため、同一のデータセットに所属するサブセットについて言うと、「全てが metadata_file オプションを持つ」か「全てが metadata_file オプションを持たない」かのどちらかになっていれば問題ありません。

以下、利用可能なオプションを説明します。コマンドライン引数と名称が同一のオプションについては、基本的に説明を割愛します。他の README を参照してください。

全学習方法で共通のオプション

学習方法によらずに指定可能なオプションです。

データセット向けオプション

データセットの設定に関わるオプションです。datasets.subsets には記述できません。

オプション名 設定例 [general] [[datasets]]
batch_size 1 o o
bucket_no_upscale true o o
bucket_reso_steps 64 o o
enable_bucket true o o
max_bucket_reso 1024 o o
min_bucket_reso 128 o o
resolution 256, [512, 512] o o
  • batch_size
    • コマンドライン引数の --train_batch_size と同等です。

これらの設定はデータセットごとに固定です。 つまり、データセットに所属するサブセットはこれらの設定を共有することになります。 例えば解像度が異なるデータセットを用意したい場合は、上に挙げた例のように別々のデータセットとして定義すれば別々の解像度を設定可能です。

サブセット向けオプション

サブセットの設定に関わるオプションです。

オプション名 設定例 [general] [[datasets]] [[dataset.subsets]]
color_aug false o o o
face_crop_aug_range [1.0, 3.0] o o o
flip_aug true o o o
keep_tokens 2 o o o
num_repeats 10 o o o
random_crop false o o o
shuffle_caption true o o o
caption_prefix “masterpiece, best quality, ” o o o
caption_suffix “, from side” o o o
caption_separator (通常は設定しません) o o o
keep_tokens_separator `“ ”`
secondary_separator “;;;” o o o
enable_wildcard true o o o
  • num_repeats

    • サブセットの画像の繰り返し回数を指定します。fine tuning における --dataset_repeats に相当しますが、num_repeats はどの学習方法でも指定可能です。
  • caption_prefix, caption_suffix

    • キャプションの前、後に付与する文字列を指定します。シャッフルはこれらの文字列を含めた状態で行われます。keep_tokens を指定する場合には注意してください。
  • caption_separator

    • タグを区切る文字列を指定します。デフォルトは , です。このオプションは通常は設定する必要はありません。
  • keep_tokens_separator

    • キャプションで固定したい部分を区切る文字列を指定します。たとえば aaa, bbb ||| ccc, ddd, eee, fff ||| ggg, hhh のように指定すると、aaa, bbbggg, hhh の部分はシャッフル、drop されず残ります。間のカンマは不要です。結果としてプロンプトは aaa, bbb, eee, ccc, fff, ggg, hhhaaa, bbb, fff, ccc, eee, ggg, hhh などになります。
  • secondary_separator

    • 追加の区切り文字を指定します。この区切り文字で区切られた部分は一つのタグとして扱われ、シャッフル、drop されます。その後、caption_separator に置き換えられます。たとえば aaa;;;bbb;;;ccc のように指定すると、aaa,bbb,ccc に置き換えられるか、まとめて drop されます。
  • enable_wildcard

    • ワイルドカード記法および複数行キャプションを有効にします。ワイルドカード記法、複数行キャプションについては後述します。

DreamBooth 方式専用のオプション

DreamBooth 方式のオプションは、サブセット向けオプションのみ存在します。

サブセット向けオプション

DreamBooth 方式のサブセットの設定に関わるオプションです。

オプション名 設定例 [general] [[datasets]] [[dataset.subsets]]
image_dir ‘C:\hoge’ - - o(必須)
caption_extension ".txt" o o o
class_tokens “sks girl” - - o
cache_info false o o o
is_reg false - - o

まず注意点として、 image_dir には画像ファイルが直下に置かれているパスを指定する必要があります。従来の DreamBooth の手法ではサブディレクトリに画像を置く必要がありましたが、そちらとは仕様に互換性がありません。また、5_cat のようなフォルダ名にしても、画像の繰り返し回数とクラス名は反映されません。これらを個別に設定したい場合、num_repeatsclass_tokens で明示的に指定する必要があることに注意してください。

  • image_dir
    • 画像ディレクトリのパスを指定します。指定必須オプションです。
    • 画像はディレクトリ直下に置かれている必要があります。
  • class_tokens
    • クラストークンを設定します。
    • 画像に対応する caption ファイルが存在しない場合にのみ学習時に利用されます。利用するかどうかの判定は画像ごとに行います。class_tokens を指定しなかった場合に caption ファイルも見つからなかった場合にはエラーになります。
  • cache_info
    • 画像サイズ、キャプションをキャッシュするかどうかを指定します。指定しなかった場合は false になります。キャッシュは image_dirmetadata_cache.json というファイル名で保存されます。
    • キャッシュを行うと、二回目以降のデータセット読み込みが高速化されます。数千枚以上の画像を扱う場合には有効です。
  • is_reg
    • サブセットの画像が正規化用かどうかを指定します。指定しなかった場合は false として、つまり正規化画像ではないとして扱います。

fine tuning 方式専用のオプション

fine tuning 方式のオプションは、サブセット向けオプションのみ存在します。

サブセット向けオプション

fine tuning 方式のサブセットの設定に関わるオプションです。

オプション名 設定例 [general] [[datasets]] [[dataset.subsets]]
image_dir ‘C:\hoge’ - - o
metadata_file 'C:\piyo\piyo_md.json' - - o(必須)
  • image_dir
    • 画像ディレクトリのパスを指定します。DreamBooth の手法の方とは異なり指定は必須ではありませんが、設定することを推奨します。
      • 指定する必要がない状況としては、メタデータファイルの生成時に --full_path を付与して実行していた場合です。
    • 画像はディレクトリ直下に置かれている必要があります。
  • metadata_file
    • サブセットで利用されるメタデータファイルのパスを指定します。指定必須オプションです。
      • コマンドライン引数の --in_json と同等です。
    • サブセットごとにメタデータファイルを指定する必要がある仕様上、ディレクトリを跨いだメタデータを1つのメタデータファイルとして作成することは避けた方が良いでしょう。画像ディレクトリごとにメタデータファイルを用意し、それらを別々のサブセットとして登録することを強く推奨します。

caption dropout の手法が使える場合に指定可能なオプション

caption dropout の手法が使える場合のオプションは、サブセット向けオプションのみ存在します。 DreamBooth 方式か fine tuning 方式かに関わらず、caption dropout に対応している学習方法であれば指定可能です。

サブセット向けオプション

caption dropout が使えるサブセットの設定に関わるオプションです。

オプション名 [general] [[datasets]] [[dataset.subsets]]
caption_dropout_every_n_epochs o o o
caption_dropout_rate o o o
caption_tag_dropout_rate o o o

重複したサブセットが存在する時の挙動

DreamBooth 方式のデータセットの場合、その中にある image_dir が同一のサブセットは重複していると見なされます。 fine tuning 方式のデータセットの場合は、その中にある metadata_file が同一のサブセットは重複していると見なされます。 データセット中に重複したサブセットが存在する場合、2個目以降は無視されます。

一方、異なるデータセットに所属している場合は、重複しているとは見なされません。 例えば、以下のように同一の image_dir を持つサブセットを別々のデータセットに入れた場合には、重複していないと見なします。 これは、同じ画像でも異なる解像度で学習したい場合に役立ちます。

# 別々のデータセットに存在している場合は重複とは見なされず、両方とも学習に使われる

[[datasets]]
resolution = 512

  [[datasets.subsets]]
  image_dir = 'C:\hoge'

[[datasets]]
resolution = 768

  [[datasets.subsets]]
  image_dir = 'C:\hoge'

コマンドライン引数との併用

設定ファイルのオプションの中には、コマンドライン引数のオプションと役割が重複しているものがあります。

以下に挙げるコマンドライン引数のオプションは、設定ファイルを渡した場合には無視されます。

  • --train_data_dir
  • --reg_data_dir
  • --in_json

以下に挙げるコマンドライン引数のオプションは、コマンドライン引数と設定ファイルで同時に指定された場合、コマンドライン引数の値よりも設定ファイルの値が優先されます。特に断りがなければ同名のオプションとなります。

コマンドライン引数のオプション 優先される設定ファイルのオプション
--bucket_no_upscale
--bucket_reso_steps
--caption_dropout_every_n_epochs
--caption_dropout_rate
--caption_extension
--caption_tag_dropout_rate
--color_aug
--dataset_repeats num_repeats
--enable_bucket
--face_crop_aug_range
--flip_aug
--keep_tokens
--min_bucket_reso
--random_crop
--resolution
--shuffle_caption
--train_batch_size batch_size

エラーの手引き

現在、外部ライブラリを利用して設定ファイルの記述が正しいかどうかをチェックしているのですが、整備が行き届いておらずエラーメッセージがわかりづらいという問題があります。 将来的にはこの問題の改善に取り組む予定です。

次善策として、頻出のエラーとその対処法について載せておきます。 正しいはずなのにエラーが出る場合、エラー内容がどうしても分からない場合は、バグかもしれないのでご連絡ください。

  • voluptuous.error.MultipleInvalid: required key not provided @ ...: 指定必須のオプションが指定されていないというエラーです。指定を忘れているか、オプション名を間違って記述している可能性が高いです。
    • ... の箇所にはエラーが発生した場所が載っています。例えば voluptuous.error.MultipleInvalid: required key not provided @ data['datasets'][0]['subsets'][0]['image_dir'] のようなエラーが出たら、0 番目の datasets 中の 0 番目の subsets の設定に image_dir が存在しないということになります。
  • voluptuous.error.MultipleInvalid: expected int for dictionary value @ ...: 指定する値の形式が不正というエラーです。値の形式が間違っている可能性が高いです。int の部分は対象となるオプションによって変わります。この README に載っているオプションの「設定例」が役立つかもしれません。
  • voluptuous.error.MultipleInvalid: extra keys not allowed @ ...: 対応していないオプション名が存在している場合に発生するエラーです。オプション名を間違って記述しているか、誤って紛れ込んでいる可能性が高いです。

その他

複数行キャプション

enable_wildcard = true を設定することで、複数行キャプションも同時に有効になります。キャプションファイルが複数の行からなる場合、ランダムに一つの行が選ばれてキャプションとして利用されます。

1girl, hatsune miku, vocaloid, upper body, looking at viewer, microphone, stage
a girl with a microphone standing on a stage
detailed digital art of a girl with a microphone on a stage

ワイルドカード記法と組み合わせることも可能です。

メタデータファイルでも同様に複数行キャプションを指定することができます。メタデータの .json 内には、\n を使って改行を表現してください。キャプションファイルが複数行からなる場合、merge_captions_to_metadata.py を使うと、この形式でメタデータファイルが作成されます。

メタデータのタグ (tags) は、キャプションの各行に追加されます。

{
    "/path/to/image.png": {
        "caption": "a cartoon of a frog with the word frog on it\ntest multiline caption1\ntest multiline caption2",
        "tags": "open mouth, simple background, standing, no humans, animal, black background, frog, animal costume, animal focus"
    },
    ...
}

この場合、実際のキャプションは a cartoon of a frog with the word frog on it, open mouth, simple background ... または test multiline caption1, open mouth, simple background ...test multiline caption2, open mouth, simple background ... 等になります。

設定ファイルの記述例:追加の区切り文字、ワイルドカード記法、keep_tokens_separator

[general]
flip_aug = true
color_aug = false
resolution = [1024, 1024]

[[datasets]]
batch_size = 6
enable_bucket = true
bucket_no_upscale = true
caption_extension = ".txt"
keep_tokens_separator= "|||"
shuffle_caption = true
caption_tag_dropout_rate = 0.1
secondary_separator = ";;;" # subset 側に書くこともできます / can be written in the subset side
enable_wildcard = true # 同上 / same as above

  [[datasets.subsets]]
  image_dir = "/path/to/image_dir"
  num_repeats = 1

  # ||| の前後はカンマは不要です(自動的に追加されます) / No comma is required before and after ||| (it is added automatically)
  caption_prefix = "1girl, hatsune miku, vocaloid |||" 
  
  # ||| の後はシャッフル、drop されず残ります / After |||, it is not shuffled or dropped and remains
  # 単純に文字列として連結されるので、カンマなどは自分で入れる必要があります / It is simply concatenated as a string, so you need to put commas yourself
  caption_suffix = ", anime screencap ||| masterpiece, rating: general"

キャプション記述例、secondary_separator 記法:secondary_separator = ";;;" の場合

1girl, hatsune miku, vocaloid, upper body, looking at viewer, sky;;;cloud;;;day, outdoors

sky;;;cloud;;;day の部分はシャッフル、drop されず sky,cloud,day に置換されます。シャッフル、drop が有効な場合、まとめて(一つのタグとして)処理されます。つまり vocaloid, 1girl, upper body, sky,cloud,day, outdoors, hatsune miku (シャッフル)や vocaloid, 1girl, outdoors, looking at viewer, upper body, hatsune miku (drop されたケース)などになります。

キャプション記述例、ワイルドカード記法: enable_wildcard = true の場合

1girl, hatsune miku, vocaloid, upper body, looking at viewer, {simple|white} background

ランダムに simple または white が選ばれ、simple background または white background になります。

1girl, hatsune miku, vocaloid, {{retro style}}

タグ文字列に {} そのものを含めたい場合は {{}} のように二つ重ねてください(この例では実際に学習に用いられるキャプションは {retro style} になります)。

キャプション記述例、keep_tokens_separator 記法: keep_tokens_separator = "|||" の場合

1girl, hatsune miku, vocaloid ||| stage, microphone, white shirt, smile ||| best quality, rating: general

1girl, hatsune miku, vocaloid, microphone, stage, white shirt, best quality, rating: general1girl, hatsune miku, vocaloid, white shirt, smile, stage, microphone, best quality, rating: general などになります。