feat: 色情图片识别(NudeNet);明暗档位改为5档(20%~100%)

This commit is contained in:
2025-06-02 21:04:38 +08:00
parent ea88f3e766
commit a914b4be74
13 changed files with 450 additions and 25 deletions
+14 -2
View File
@@ -58,10 +58,22 @@ func (o Orientation) String() string {
// - 「图片_GIF」:GIF 动图单独放(不做明暗分类)
// - 「图片_横屏_1080P/bright_20」:普通图片 = 方向 + 清晰度,明暗作为其子目录
func ImageDir(ext string, width, height int, darkScore, lightScore float64) string {
return imageDir("图片", ext, width, height, darkScore, lightScore)
}
// ImageDirSensitive 返回一张被判为色情的图片该进的文件夹路径,比如:
// - 「图片_敏感_GIF」:GIF 单独放(不做明暗分类)
// - 「图片_敏感_横屏_1080P/bright_20」:普通分类前加「_敏感」前缀,明暗仍是子目录
func ImageDirSensitive(ext string, width, height int, darkScore, lightScore float64) string {
return imageDir("图片_敏感", ext, width, height, darkScore, lightScore)
}
// imageDir 拼出"类型 + 方向 + 清晰度 + 明暗子目录"这段路径。
func imageDir(kind, ext string, width, height int, darkScore, lightScore float64) string {
if strings.EqualFold(ext, ".gif") {
return "图片_GIF"
return kind + "_GIF"
}
dir := resolutionDir("图片", ext, width, height)
dir := resolutionDir(kind, ext, width, height)
return dir + "/" + BrightnessName(darkScore, lightScore)
}