WD 1.4 ConvNextV2 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.3710, F1 = 0.6862

最后说明

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