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
+8 -8
View File
@@ -7,12 +7,12 @@ import "fmt"
const neutralMargin = 0.15
// BrightnessName 根据暗色/亮色像素占比,给图片定"明暗等级",返回类似:
// - "dark_70" → 暗色像素占 65%~75% 的暗图
// - "bright_40" → 亮色像素占 35%~45% 的亮图
// - "dark_80" → 暗色像素占 70%~90% 的暗图
// - "bright_40" → 亮色像素占 30%~50% 的亮图
// - "mid" → 明暗差不多,不好说谁多,归中性
//
// 判断规则(大白话):谁占得多、而且多出明显一截(超过 neutralMargin),
// 就按谁命名,数字是占比取整到 10% 一档。
// 就按谁命名,数字是占比取整到 20% 一档。
func BrightnessName(darkScore, lightScore float64) string {
switch {
case lightScore > darkScore+neutralMargin:
@@ -24,13 +24,13 @@ func BrightnessName(darkScore, lightScore float64) string {
}
}
// brightnessTier 把占比(0~1)取整到"最接近的 10%"作为档位数字,范围 10~100。
// 例:0.30~0.39 → "30";0.96 以上 → "100"。
// brightnessTier 把占比(0~1)取整到"最接近的 20%"作为档位数字,范围 20~100。
// 例:0.30~0.50 → "40";0.96 以上 → "100"。
// 这样文件夹名里的数字不会出现"明明很少却挂个大数字"的情况。
func brightnessTier(score float64) int {
tier := int(score*100+5) / 10 * 10
if tier < 10 {
return 10
tier := int(score*100+10) / 20 * 20
if tier < 20 {
return 20
}
if tier > 100 {
return 100