WD 1.4 ViT Tagger V2

支持评级、角色和一般标签。

使用 https://github.com/SmilingWolf/SW-CV-ModelZoo 进行训练。训练使用的TPU由 TRC program 友情提供。

数据集

最后图像ID：5944504，训练数据集为具有模数0000-0899的Danbooru图像。验证数据集为具有模数0950-0999的图像。过滤掉具有少于10个一般标签的图像。过滤掉具有少于600个图像的标签。

验证结果

P=R: 阈值 = 0.3537，F1 = 0.6770

总结

可能会有变化和更新。鼓励下游用户使用已标记的发行版本，而不是依赖存储库的最新版本。