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
+12 -2
View File
@@ -14,11 +14,21 @@ organize.rotateImages=false
organize.workers=8
# 色情图片识别(需本机安装 python 与 nudenet:pip install nudenet)
# true=开启,判为色情的图片移入「图片_敏感_*」目录(明暗子目录照常);false=关闭(默认)
# 模型:优先加载程序运行目录下 lib/320n.onnx(推荐随程序分发,离线可用);
# 模型:优先加载程序运行目录下 lib/nwfs/320n.onnx(推荐随程序分发,离线可用);
# 本地没有会自动下载到该位置,下载失败再回退用 nudenet 自带模型
# NSFW是用来识别真人图片的
nsfw.enabled=true
nsfw.enabled=false
# python 可执行文件路径;留空自动在 PATH 里找 python/python3/py
nsfw.python=C:\\Users\\Administrator\\AppData\\Local\\Programs\\Python\\Python312\\python.exe
# 判定阈值 0~1,敏感部位检测分高于此才算色情(默认 0.2,实测)
nsfw.threshold=0.2
# 二次元图片识别(需本机安装 python 与 onnxruntime、Pillow、numpy:pip install onnxruntime Pillow numpy)
# true=开启,判为二次元敏感(questionable/explicit)的图片移入「图片_二次元_*」目录;false=关闭(默认)
# 模型:首次启动会自动从 HuggingFace 下载 wd-vit-large-tagger-v3 到 lib/wd-tagger/(约 1.5GB)
# Hentai是用来识别二次元/动漫风格图片的
hentai.enabled=true
# python 可执行文件路径;留空自动在 PATH 里找 python/python3/py(和 nsfw.python 共用同一个即可)
hentai.python=C:\\Users\\Administrator\\AppData\\Local\\Programs\\Python\\Python312\\python.exe
# 判定阈值 0~1,questionable 或 explicit 分数高于此才算二次元敏感(默认 0.5)
hentai.threshold=0.5