kohya-ss 産 WD14Tagger を使用してみましたが、
先日の調査によると、他にもコマンドラインから使用できるようにしてくれている方がいました。
本日は corkborg/wd14-tagger-standalone を使ってみようと思います!
教師データ:ずんだもんの準備
ずんだもん は Google Drive に置きっぱなしになっているいつものを使用します。
from google.colab import drive drive.mount('/content/drive')
!cp -r /content/drive/MyDrive/trains/dataset .
!rm /content/dataset/zundamon/*.txt
準備の詳細は LoRA を自作したい - 準備するもの (2)#ディレクトリ構成を作って Google Drive へ をご参照ください。
リポジトリをクローンしてくる
!git clone https://github.com/corkborg/wd14-tagger-standalone.git
実行準備
%cd wd14-tagger-standalone
!pip install -U -r requirements.txt !pip install onnx onnxruntime-gpu
実行!
!python run.py --dir /content/dataset/zundamon --model wd-vit-large-tagger-v3 processing: /content/dataset/zundamon/zundamon (10).png Loading WD ViT-Large Tagger v3 model file from SmilingWolf/wd-vit-large-tagger-v3 model.onnx: 100% 1.26G/1.26G [00:08<00:00, 154MB/s] selected_tags.csv: 100% 308k/308k [00:00<00:00, 36.7MB/s] 2024-08-28 12:49:04.450220806 [E:onnxruntime:Default, provider_bridge_ort.cc:1992 TryGetProviderInfo_CUDA] /onnxruntime_src/onnxruntime/core/session/provider_bridge_ort.cc:1637 onnxruntime::Provider& onnxruntime::ProviderLibrary::Get() [ONNXRuntimeError] : 1 : FAIL : Failed to load library libonnxruntime_providers_cuda.so with error: libcudnn.so.9: cannot open shared object file: No such file or directory 2024-08-28 12:49:04.450256259 [W:onnxruntime:Default, onnxruntime_pybind_state.cc:965 CreateExecutionProviderInstance] Failed to create CUDAExecutionProvider. Require cuDNN 9.* and CUDA 12.*. Please install all dependencies as mentioned in the GPU requirements page (https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements), make sure they're in the PATH, and that your GPU is supported. Loaded WD ViT-Large Tagger v3 model from /root/.cache/huggingface/hub/models--SmilingWolf--wd-vit-large-tagger-v3/snapshots/ae469aa2e4706a3af08d3673cf73a11d1add314c/model.onnx processing: /content/dataset/zundamon/zundamon (2).png processing: /content/dataset/zundamon/zundamon (1).png processing: /content/dataset/zundamon/zundamon (3).png processing: /content/dataset/zundamon/zundamon (7).png processing: /content/dataset/zundamon/zundamon (12).png processing: /content/dataset/zundamon/zundamon (11).png processing: /content/dataset/zundamon/zundamon (8).png processing: /content/dataset/zundamon/zundamon (5).png processing: /content/dataset/zundamon/zundamon (4).png processing: /content/dataset/zundamon/zundamon (6).png processing: /content/dataset/zundamon/zundamon (9).png
まとめ
なんかいい感じに動いてくれてる予感!