WD 1.4 MOAT标记器 V2

支持评级、角色和常规标签。

使用 https://github.com/SmilingWolf/SW-CV-ModelZoo 进行训练。TPUs训练的友好提供者是 TRC program 。

数据集

最后一个图像的ID为5944504。使用模ID为0000-0899的Danbooru图像进行训练。使用模ID为0950-0999的图像进行验证。过滤掉少于10个常规标签的图像。过滤掉少于600个图像的标签。

验证结果

P = R: 阈值为0.3771，F1 = 0.6911

论文

MOAT: Alternating Mobile Convolution and Attention Brings Strong Vision Models

最后一句话

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