File size: 1,713 Bytes
cfdc26c 28f31d2 66af87a 3a1bbed bdc6ca1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
---
license: mit
datasets:
- deepghs/monochrome_danbooru
pipeline_tag: image-classification
library_name: dghs-imgutils
tags:
- art
- onnx
- safetensors
- emb_attachments
- wdtaggers
---
This is an experiment on DeepGHS's new tagger attachment training framework [deepghs/emb_attachments](https://github.com/deepghs/emb_attachments).
Just some technical validation, not recommended for use in production.
(2025.3.7: This feature is still WIP, you can take a look at [this PR](https://github.com/deepghs/imgutils/pull/152) and [this documentation](https://dghs-imgutils.deepghs.org/dev/attachments/api_doc/tagging/wd14.html#get-wd14-tags), you will like it.)
| Name | Encoder | Problem | FLOPS | Params | AUC | accuracy | loss | mAP | Labels |
|:---------------------------:|:----------------------------------------:|:--------------:|:-------:|:--------:|:------:|:----------:|:-------:|:------:|:------------------:|
| mlp_layer1_seed114514_dist8 | wdtagger:SmilingWolf/wd-swinv2-tagger-v3 | classification | 525.3K | 525.8K | 0.9937 | 0.9743 | 0.04483 | 0.9937 | monochrome, normal |
| mlp_layer1_seedx_dist8 | wdtagger:SmilingWolf/wd-swinv2-tagger-v3 | classification | 525.3K | 525.8K | 0.9946 | 0.9732 | 0.04058 | 0.9947 | monochrome, normal |
| mlp_layer1_seed3_dist8 | wdtagger:SmilingWolf/wd-swinv2-tagger-v3 | classification | 525.3K | 525.8K | 0.9956 | 0.9699 | 0.05301 | 0.9957 | monochrome, normal |
| mlp_layer1_seed1 | wdtagger:SmilingWolf/wd-swinv2-tagger-v3 | classification | 525.3K | 525.8K | 0.9948 | 0.9652 | 0.06904 | 0.995 | monochrome, normal |
|