WD 1.4 MOAT Tagger V2

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

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

数据集

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

验证结果

P=R: threshold = 0.3771, F1 = 0.6911

论文

MOAT: Alternating Mobile Convolution and Attention Brings Strong Vision Models

最后说明

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