feat: 二次元识别功能 + 模型目录整理(nwfs/wd-tagger 子目录),修复 wd-tagger 输入维度(NHWC)

This commit is contained in:
2025-06-25 21:10:21 +08:00
parent 4efa48f550
commit 3af2bf70b4
18 changed files with 1386 additions and 21 deletions
+1 -1
View File
@@ -19,7 +19,7 @@
|------|------|------|
| 真人 vs 动漫 | `dghs-imgutils` `anime_real` | 二分类,输出 real/anime 概率 |
| 真人 vs 风景 | 真人检测器 | 检测到人脸/人物 → 真人,否则风景 |
| 真人鉴黄 | NudeNet(已有 `lib/320n.onnx`) | 保持现状 |
| 真人鉴黄 | NudeNet(已有 `lib/nwfs/320n.onnx`) | 保持现状 |
| 动漫鉴黄 | `dghs-imgutils` `anime_rating` / `nsfw_pred` | danbooru 评级:safe/questionable/explicit 分数 |
- `dghs-imgutils`(`pip install dghs-imgutils`)一个包覆盖「真人/动漫判定」+「动漫鉴黄」+「动漫分类」,