| 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 | | |