You're looking at a specific version of this model. Jump to the model overview.

zsxkib /wd-image-tagger:2c081a1d

Playground API

Input schema

The fields you can use to run this model with an API. If you don’t give a value for a field its default value will be used.

Field Type Default value Description
image
string
Path to the input image file to be analyzed by the WaifuDiffusion tagger model
model_repo
None
models/wd-swinv2-tagger-v3
Name of the pre-trained model repository to use for image analysis
general_thresh
number
0.35

Max: 1

Probability threshold for including general tags in the output (between 0 and 1)
general_mcut_enabled
boolean
False
Whether to use the MCut algorithm to automatically determine the general tags threshold
character_thresh
number
0.85

Max: 1

Probability threshold for including character tags in the output (between 0 and 1)
character_mcut_enabled
boolean
False
Whether to use the MCut algorithm to automatically determine the character tags threshold
category
None
all_tags
Category of tags to return in the output

Output schema

The shape of the response you’ll get when you run this model with an API.

Schema 
{'items': {'properties': {'category': {'title': 'Category', 'type': 'string'},
                          'confidence': {'title': 'Confidence',
                                         'type': 'number'},
                          'tag': {'title': 'Tag', 'type': 'string'}},
           'required': ['tag', 'confidence', 'category'],
           'title': 'Tag',
           'type': 'object'},
 'title': 'Output',
 'type': 'array'}