WD14Tagger

Automatically tag images with booru tags.

Requirements

This project requires cudnn 8 to properly use the GPU.

On Arch:

  sudo pacman -S cudnn

Installation

  git clone https://github.com/KutsuyaYuki/WD14Tagger
  cd WD14Tagger
  conda env create -f environment.yaml
  conda activate wd14tagger
  pip install -r requirements.txt

Usage/Examples

  python tag_images_by_wd14_tagger.py \
  input \
  --batch_size 4 \
  --caption_extension .txt

Change input to the folder where your images are located. For example, if they are located in a folder called images on your desktop:

  python tag_images_by_wd14_tagger.py \
  ~/Desktop/images \
  --batch_size 4 \
  --caption_extension .txt