Join the conversation

Join the community of Machine Learners and AI enthusiasts.

Sign Up
v2ray 
posted an update Aug 23
Post
1356
v2ray/deepgelbooru
A Danbooru tag image tagger, maybe better than WD14 at some images.
Training code, inference code, dataset included.
:3

ok but what is it ? cnn ? vit ?

·

cnn

I don't have a lot of images, so I couldn't tell the difference between the optim and the ones that don't...🙀
They are twice the size.
In any case, it is certainly a Danbooru-type tagger with performance comparable to the WD14 series.
By the way, it worked even without Zero GPU space.
https://huggingface.co/spaces/John6666/danbooru-tags-transformer-v2-with-wd-tagger-b

·

from my tests it's kind of similar to wd14 but slightly worse for finding little details, it's probably because the limitation of cnns
btw the optim one is the optimizer state, it's not a model

im also pretraining a clip model with a classification head but it will most likely be worse than this one