DeepDanbooru

收录理由

如果你正在为动漫或插画图库开发打标工具，DeepDanbooru 是一个相当成熟的选择。它基于多标签分类模型，单次推理就能为一张图片预估出数百个标签，这些标签体系源自 Danbooru 社区长期维护的标注数据。整个流程都在命令行中完成：你可以拉取最新的标签列表，筛选带有 SQLite 注释的图片数据库，用自己收集的数据训练模型，然后对单张图片或整个文件夹进行预测。自行训练模型意味着你不必受限于固定的预训练标签集，当你的图库使用不同的词汇或标注习惯时，这一点尤为重要。项目还附带了 Grad-CAM 可视化工具，能直观显示图片的哪些区域影响了某个标签的判断，在依赖结果之前可以先确认模型的依据是否合理。