WD 1.4 ConvNext Tagger V2

支持分级标签、角色标签和通用标签。

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

数据集

最新图片 ID：5944504
训练集为 Danbooru 中 ID 模运算结果在 0000-0899 范围内的图片。
验证集为 ID 模运算结果在 0950-0999 范围内的图片。
过滤掉通用标签少于 10 个的图片。
过滤掉出现次数少于 600 张图片的标签。

验证结果

P=R: threshold = 0.3685, F1 = 0.6810

结语

可能会进行更改和更新。
建议下游用户使用带标签的发布版本，而非依赖仓库的最新代码。