WD 1.4 ConvNext Tagger V2

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

使用 https://github.com/SmilingWolf/SW-CV-ModelZoo 进行训练。训练中使用的 TPUs 由 TRC program 慷慨提供。

数据集

最后图像编号：5944504。训练于图像的 ID 取模为 0000-0899。验证于图像的 ID 取模为 0950-0999。筛除了具有少于 10 个常见标签的图像。筛除了具有少于 600 张图像的标签。

验证结果

P=R: 阈值 = 0.3685，F1 = 0.6810

最后的话

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