Google Colab で kohya-ss 産 WD14Tagger 使ってみたよ

zako-lab929.hatenablog.com

こちらの記事で調べてた内容です。

結局 kohya-ss/sd-scripts 産の WD14Tagger を使うことができました

結論
- 補足
Google Colab で使ってみた
まとめ

結論

# !python finetune/tag_images_by_wd14_tagger.py --onnx --repo_id <REPO_ID> --batch_size <BATCH_SIZE> <教師データフォルダ>
!python finetune/tag_images_by_wd14_tagger.py --onnx --repo_id SmilingWolf/wd-swinv2-tagger-v3 --batch_size 4 /content/dataset/zundamon

こんな感じで使えました

補足

train_README-ja.md#WD14Taggerによるタグ付け に記載のオプションでは使うことができなかったので、

wd14_tagger_README-ja.md#WD14Taggerによるタグ付け を参考にすると良いです

Google Colab で使ってみた

教師データの準備

教師データにはずっと前から使っているずんだもんセットを使います

from google.colab import drive
drive.mount('/content/drive')

!cp -r /content/drive/MyDrive/trains/dataset .

!rm /content/dataset/zundamon/*.txt

以下で準備した ずんだもんセット を Google Drive にアップロードしてありますので、Google Drive をマウントして持ってくる形となります。

zako-lab929.hatenablog.com

但し、今回は Tagger を使って、 .txt ファイル を自動生成することが目的ですので、 ずんだもんセット のうち .txt は削除しておきます。

kohya-ss/sd-scripts リポジトリの準備

!git clone https://github.com/kohya-ss/sd-scripts.git
%cd sd-scripts

!pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121
!pip install -U -r requirements.txt
!pip install xformers==0.0.23.post1 --index-url https://download.pytorch.org/whl/cu121

tag_images_by_wd14_tagger.py を使う準備

!pip install onnx==1.15.0 onnxruntime-gpu==1.17.1

実行！！

!python finetune/tag_images_by_wd14_tagger.py --onnx --repo_id SmilingWolf/wd-swinv2-tagger-v3 --batch_size 4 /content/dataset/zundamon

出力結果！！

!ls /content/dataset/zundamon
'zundamon (10).png'  'zundamon (1).png'  'zundamon (4).png'  'zundamon (7).png'
'zundamon (10).txt'  'zundamon (1).txt'  'zundamon (4).txt'  'zundamon (7).txt'
'zundamon (11).png'  'zundamon (2).png'  'zundamon (5).png'  'zundamon (8).png'
'zundamon (11).txt'  'zundamon (2).txt'  'zundamon (5).txt'  'zundamon (8).txt'
'zundamon (12).png'  'zundamon (3).png'  'zundamon (6).png'  'zundamon (9).png'
'zundamon (12).txt'  'zundamon (3).txt'  'zundamon (6).txt'  'zundamon (9).txt'

.txt ファイルが自動生成されました！

試しに zundamon (1).txt の中身は以下のようになっています

zundamon (1).txt

1girl, solo, long_hair, looking_at_viewer, smile, open_mouth, simple_background, shirt, white_background, bow, hair_between_eyes, very_long_hair, white_shirt, full_body, yellow_eyes, short_sleeves, one_eye_closed, shorts, green_hair, shoes, socks, puffy_sleeves, hand_up, sailor_collar, arm_up, puffy_short_sleeves, v, kneehighs, black_socks, pink_bow, low_ponytail, pink_ribbon, green_shorts, green_footwear, green_sailor_collar, legwear_garter, zundamon