deepghs

/

wd14_tagger_with_embeddings

like 7

Follow

DeepGHS 189

use this: https://deepghs.github.io/imgutils/main/api_doc/tagging/wd14.html#get-wd14-tags

from imgutils.tagging import get_wd14_tags

embeddings = get_wd14_tags(
    'waifu_image.png',
    # model_name='EVA02_Large',
    fmt='embedding',
)
print(embeddings.shape, embeddings.dtype)

library's readme: https://github.com/deepghs/imgutils

@narugo thank you! I found the same library by searching "deepghs/wd14_tagger_with_embeddings" in github