package classify import "fmt" // 暗色阈值等级(从高到低) var darkThresholds = []int{100, 95, 85, 75, 65, 55} // 亮色阈值等级(从高到低) var brightThresholds = []int{85, 75, 65, 55, 45, 35, 25, 15} // BrightnessName 按主导分判定明暗等级,返回如 "dark_100" / "bright_85"。 // 亮色主导(lightScore > darkScore)归 bright,否则归 dark,每张图必归亮暗之一。 func BrightnessName(darkScore, lightScore float64) string { darkLowest := darkThresholds[len(darkThresholds)-1] brightLowest := brightThresholds[len(brightThresholds)-1] if lightScore > darkScore { picked := brightLowest for _, t := range brightThresholds { if lightScore >= float64(t)/100.0 { picked = t break } } return fmt.Sprintf("bright_%d", picked) } picked := darkLowest for _, t := range darkThresholds { if darkScore >= float64(t)/100.0 { picked = t break } } return fmt.Sprintf("dark_%d", picked) }