wbz666666/wd-v1-4-convnext-tagger-v2

By wbz666666 · ↓ 11 · ♥ 0 · Updated 2026-03-09

What's in this repo

This page lists the files in wbz666666/wd-v1-4-convnext-tagger-v2, including model.onnx.

1 ONNX models (.onnx files)
8 files, including 2 .pb, 1 .md, 1 .onnx, 1 .csv, 1 .data-00000-of-00001, 1 .index
Last updated 2026-03-09

ONNX files

File	Size	Download
model.onnx	—	Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

WD 1.4 ConvNext Tagger V2

Supports ratings, characters and general tags.

Trained using https://github.com/SmilingWolf/SW-CV-ModelZoo.
TPUs used for training kindly provided by the TRC program.

Dataset

Last image id: 5944504
Trained on Danbooru images with IDs modulo 0000-0899.
Validated on images with IDs modulo 0950-0999.
Images with less than 10 general tags were filtered out.
Tags with less than 600 images were filtered out.

Validation results

P=R: threshold = 0.3685, F1 = 0.6810

Final words

Subject to change and updates.
Downstream users are encouraged to use tagged releases rather than relying on the head of the repo.

Model info

tf-keras

tf-keras onnx license:apache-2.0 region:us

↓ 11 ♥ 0

Created: 2026-03-09

Updated: 2026-03-09

View on Hugging Face

Files (8)

.gitattributes

README.md

keras_metadata.pb

model.onnx ONNX

saved_model.pb

selected_tags.csv

variables/variables.data-00000-of-00001

variables/variables.index

Related models

More from wbz666666

No other models by this author.

More models

No task type declared.