Tag images
Project description
img2tags
Setup
# For CPU
pip install 'img2tags[cpu]'
# For GPU
pip install 'img2tags[gpu]'
Run
# Generate .txt file
img2tags -i input_dir
# Generate .json file in another directory with CPU
img2tags --model shirayu/img2tags/SmilingWolf__wd-v1-4-moat-tagger-v2 \
--batch 4 \
--ext json \
--cpu \
-i input_dir \
-o output_dir
LICENSE
Apache 2.0
Some codes are based on codes in the following project.
References
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
img2tags-1.0.0.tar.gz
(8.7 kB
view hashes)