zsxkib / wd-image-tagger

Image tagger fine-tuned on WaifuDiffusion w/ (SwinV2, SwinV2, ConvNext, and ViT)

  • Public
  • 1K runs
  • T4
  • GitHub
  • License
Run with an API
  • Prediction

    zsxkib/wd-image-tagger:f9cfa5bef6e8f789662b1d6014a378f5e514a9f4d6b409e0ad84da126a4bdf62
    ID
    q2x669wp75rgc0cfmr4r6pz7m8
    Status
    Succeeded
    Source
    Web
    Hardware
    T4
    Total duration
    Created

    Input

    image
    image
    category
    all_tags
    model_repo
    models/wd-swinv2-tagger-v3
    general_thresh
    0.35
    character_thresh
    0.85
    general_mcut_enabled
    character_mcut_enabled

    Output

    tag

    outdoors

    category

    general

    confidence

    0.5684077143669128

    tag

    lying

    category

    general

    confidence

    0.3831140398979187

    tag

    blurry

    category

    general

    confidence

    0.8763853311538696

    tag

    no humans

    category

    general

    confidence

    0.9804705381393433

    tag

    blurry background

    category

    general

    confidence

    0.5741121172904968

    tag

    depth of field

    category

    general

    confidence

    0.5820473432540894

    tag

    animal

    category

    general

    confidence

    0.8318572640419006

    tag

    cat

    category

    general

    confidence

    0.6375629305839539

    tag

    grass

    category

    general

    confidence

    0.5245271325111389

    tag

    realistic

    category

    general

    confidence

    0.5816801190376282

    tag

    animal focus

    category

    general

    confidence

    0.9442822933197021

    tag

    white fur

    category

    general

    confidence

    0.36753329634666443

    tag

    whiskers

    category

    general

    confidence

    0.551397442817688

    tag

    general

    category

    rating

    confidence

    0.983599066734314

    tag

    sensitive

    category

    rating

    confidence

    0.015072911977767944

    tag

    questionable

    category

    rating

    confidence

    0.000958561897277832

    tag

    explicit

    category

    rating

    confidence

    0.000261843204498291
    Generated in

Want to make some of these yourself?

Run this model