Skip to main content

Tag images

Project description

img2tags

PyPI version Python Versions License Downloads

CI Typos

Tag images by using ONNX Runtime. The current default model is the converted WD 1.4 ConvNext Tagger V2 on https://huggingface.co/shirayu/img2tags. It works fine for anime images.

Setup

# For CPU
pip install -U 'img2tags[cpu]'

# For GPU
pip install -U 'img2tags[gpu]'

# If you use CUDA 12.2, the following packages may be needed
sudo apt install libcufft10 libcudart11.0

Run

# Generate .txt file
img2tags -i input_dir

# Generate .json file in another directory with CPU
#   --offline: Do not fetch model files (You must have downloaded that model)
img2tags --ext json \
    --cpu \
    -i input_dir \
    -o output_dir \
    --offline

# Output to a single file (JSONL)
img2tags --ext jsonl \
    --cpu \
    -i input_dir \
    -o output_filename.jsonl \
    --offline

# You can use filelist
img2tags --ext jsonl \
    --cpu \
    -i target_file_list.txt \
    -o output_filename.jsonl \
    --offline

The values of default thresholds are stored in config.json in each model like this. You can set values in JSON format like --th {"0": 0.3, "4":0.2, "9":0.3}.

LICENSE

Apache 2.0

Some codes are based on codes in the following project.

References

Project details


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.3.3.tar.gz (10.0 kB view hashes)

Uploaded Source

Built Distribution

img2tags-1.3.3-py3-none-any.whl (10.9 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page