DeepDanbooru とやらを使ってみる

DeepDanbooru はタグ付けを行ってくれるツールのようです。

DeepDanbooru is anime-style girl image tag estimation system.

DeepDanbooru はアニメ風の女の子画像画像タグ推定システム とのことです。

ずんだもんのデータセットを使用します
DeepDanbooru の準備する
- DeepDanbooru をクローンする
- 必要モジュールをインストールする
DeepDanbooru 用のモデルを準備する
- DeepDanbooru 用のモデルをダウンロードする
- zip を解凍する
DeepDanbooru を実行する
実行結果
まとめ

作業はいつも通り Google Colab 上で行います。

ずんだもんのデータセットを使用します

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

!cp -r /content/drive/MyDrive/trains/dataset .
!rm ./dataset/zundamon/*.txt

ここ最近使わせてもらっているセットです。
今回はキャプション出力なので *.txt は削除しておきます

DeepDanbooru の準備する

DeepDanbooru をクローンする

!git clone https://github.com/KichangKim/DeepDanbooru.git

$cd DeepDanbooru

作業ディレクトリを DeepDanbooru/ に移動します

必要モジュールをインストールする

!pip install -r requirements.txt

!pip install .[tensorflow]

DeepDanbooru 用のモデルを準備する

github.com

GitHub リポジトリ上の Release にあるモデルを使用します。

DeepDanbooru 用のモデルをダウンロードする

!wget "https://github.com/KichangKim/DeepDanbooru/releases/download/v3-20211112-sgd-e28/deepdanbooru-v3-20211112-sgd-e28.zip"

zip を解凍する

!unzip ./deepdanbooru-v3-20211112-sgd-e28.zip -d deepdanbooru-v3-20211112-sgd-e28

ここで昨日の -d です！

Ubuntu で zip 解凍時にディレクトリに出力する

DeepDanbooru を実行する

!deepdanbooru evaluate /content/dataset/zundamon --project-path deepdanbooru-v3-20211112-sgd-e28 --allow-folder --save-txt

先ほどダウンロードしてきたものは DeepDanbooru プロジェクトになっているようです

実行結果

Tags of /content/dataset/zundamon/zundamon (1).png:
1/1 ━━━━━━━━━━━━━━━━━━━━ 6s 6s/step
(0.956) 1boy
(0.620) ;d
(0.716) arm_up
(0.689) black_legwear
(0.537) brown_eyes
(0.894) full_body
(0.906) green_hair
(0.917) male_focus
(0.975) one_eye_closed
(0.955) open_mouth
(0.791) sailor_collar
(0.571) shirt
(0.839) shoes
(0.519) short_hair
(0.740) short_sleeves
(0.757) shorts
(0.852) simple_background
(0.928) smile
(0.973) solo
(0.612) standing
(0.987) v
(0.780) white_background
(0.998) rating:safe
Saved text file.

～省略～

!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'

とりあえず上手く出力できているみたいです