forked from https://github.com/picobyte/stable-diffusion-webui-wd14-tagger

install

python -m venv venv
source venv/bin/activate
pip install -r requirements.txt

usage

usage: run.py [-h] (--dir DIR | --file FILE) [--threshold THRETHOLD] [--ext EXT]
              [--model {wd14-vit.v1,wd14-vit.v2,wd14-convnext.v1,wd14-convnext.v2,wd14-convnextv2.v1,wd14-swinv2-v1,wd-v1-4-moat-tagger.v2,mld-caformer.dec-5-97527,mld-tresnetd.6-30000}]

options:
  -h, --help            show this help message and exit
  --dir DIR             ディレクトリ内の画像すべてに予測を行う
  --file FILE           ファイルに対して予測を行う
  --threshold THRETHOLD
                        予測値の足切り確率（デフォルトは0.35）
  --ext EXT             dirの場合にキャプションファイルにつける拡張子
  --model {wd14-vit.v1,wd14-vit.v2,wd14-convnext.v1,wd14-convnext.v2,wd14-convnextv2.v1,wd14-swinv2-v1,wd-v1-4-moat-tagger.v2,mld-caformer.dec-5-97527,mld-tresnetd.6-30000}
                        予測に使用するモデル名

single file

python run.py --file image.jpg

batch execution

python run.py --dir dir/dir

Support Models

python run.py --file image.jpg --model wd14-vit.v1
python run.py --file image.jpg --model wd14-vit.v2
python run.py --file image.jpg --model wd14-convnext.v1
python run.py --file image.jpg --model wd14-convnext.v2
python run.py --file image.jpg --model wd14-convnextv2.v1
python run.py --file image.jpg --model wd14-swinv2-v1
python run.py --file image.jpg --model wd-v1-4-moat-tagger.v2
python run.py --file image.jpg --model mld-caformer.dec-5-97527
python run.py --file image.jpg --model mld-tresnetd.6-30000