From c766a948da3d9a9fae35c24921f71e66ec0d31df Mon Sep 17 00:00:00 2001 From: Odboy Date: Thu, 5 Jun 2025 21:08:32 +0800 Subject: [PATCH] =?UTF-8?q?refactor:=20=E7=A7=BB=E9=99=A4=20nsfw=20?= =?UTF-8?q?=E4=B8=AD=E6=9C=AA=E4=BD=BF=E7=94=A8=E7=9A=84=E6=95=8F=E6=84=9F?= =?UTF-8?q?=E7=B1=BB=E6=B8=85=E5=8D=95(=E5=8D=95=E4=B8=80=E6=9D=A5?= =?UTF-8?q?=E6=BA=90=E5=9C=A8=20server.py)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/nsfw/detector.go | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/internal/nsfw/detector.go b/internal/nsfw/detector.go index 6f271ba..364ac68 100644 --- a/internal/nsfw/detector.go +++ b/internal/nsfw/detector.go @@ -35,14 +35,8 @@ import ( //go:embed server.py var serverScript []byte -// 敏感类集合：检测到这些部位"暴露"且分数达标才算色情。 -// 默认只取最明确的四类（生殖器/肛/女性乳房），避免把泳装、露背照误判。 -var sensitiveClasses = map[string]bool{ - "FEMALE_GENITALIA_EXPOSED": true, - "MALE_GENITALIA_EXPOSED": true, - "ANUS_EXPOSED": true, - "FEMALE_BREAST_EXPOSED": true, -} +// 注意：哪些敏感类才算色情，定义在 server.py 的 SENSITIVE 集合里（单一事实来源）， +// Go 侧只负责把路径发给服务、读取判定结果，不重复维护这份清单。 // detectorStartTimeout 等待 python 服务就绪（模型加载）的超时时间。 // 首次加载模型可能要十几秒，给足余量。