模型:

SmilingWolf/wd-v1-4-convnextv2-tagger-v2

英文

WD 1.4 ConvNextV2 Tagger V2

支持评分,角色和普通标签。

使用 https://github.com/SmilingWolf/SW-CV-ModelZoo 进行训练。感谢 TRC program 提供的训练用TPU。

数据集

最后一张图片的ID:5944504,训练使用的Danbooru图像的ID模式为0000-0899. 验证使用的图像的ID模式为0950-0999. 过滤掉普通标签少于10个的图像. 过滤掉少于600个图像的标签.

验证结果

P=R: 阈值 = 0.3710,F1 = 0.6862

最后的话

可能会有修改和更新. 鼓励下游用户使用已标记的版本,而不是依赖于存储库的最新版本.