WD 1.4 SwinV2 Tagger V2

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

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

数据集

最后图像ID：5944504。在ID模为0000-0899的Danbooru图像上进行训练。在ID模为0950-0999的图像上进行验证。过滤掉一般标签少于10个的图像。过滤掉标签图片数少于600的标签。

验证结果

P=R: 阈值=0.3771，F1 = 0.6854

最后的话

可能在将来会有变动和更新。鼓励下游用户使用已标记的版本，而不是依赖于仓库的最新版本。