/auto-tag-anime

Adds booru style tags to images by using DeepDanbooru neural net

Primary LanguagePython

Auto Anime Tag

Automatically adds booru style tags to an image or directory of images by using this neural net model: https://github.com/KichangKim/DeepDanbooru

Instructions

  1. Download the pre-trained model deepdanbooru-v3 from https://github.com/KichangKim/DeepDanbooru, put the folder containing the model in the auto-tag-anime folder

  2. python3 -m venv ./env

  3. On osx: source env/bin/activate

    On Windows: ./env/Scripts/activate

  4. python setup.py install

How to use

python3 auto-tag-anime.py "example.jpg"

python3 auto-tag-anime.py "/path/to/directory/"

Notes

  • windows version only works with jpegs
  • See a list of tags the model will predict in 'tags.txt' inside of the deepdanbooru-v3 folder
  • checks for images in subdirectories