使用AWSRekognition获取图像颜色返回nil，如何检测平均颜色？

为什么simplifiedColor、background/foreground返回nil？ #

• 调用DetectLabels接口时，默认不会返回图像属性（imageProperties）和标签实例的主色细节，必须在请求中显式指定启用这些特征。
• 需要在请求的features参数中添加.imageProperties，并通过settings.imagePropertiesConfig配置要提取的颜色信息类型。

正确配置API请求 #

以Swift为例，调整你的DetectLabels请求，确保包含图像属性检测：

let request = AWSRekognitionDetectLabelsRequest()
request.image = ... // 你的图像源（S3对象或字节数据）
request.features = [.imageProperties] // 启用图像属性检测
// 配置图像属性，指定要返回的主色数量
let imagePropertiesConfig = AWSRekognitionImagePropertiesConfig()
imagePropertiesConfig.maxDominantColors = 10 // 可根据需求调整数量
request.settings = AWSRekognitionDetectLabelsSettings()
request.settings?.imageProperties = imagePropertiesConfig

获取平均颜色的方法 #

方法1：通过主色计算加权平均 #

请求配置正确后，response.imageProperties?.dominantColors会返回图像的主色列表，每个颜色包含RGB值和像素占比。你可以基于这些数据计算加权平均颜色：

if let dominantColors = response.imageProperties?.dominantColors {
    var totalRed: Double = 0
    var totalGreen: Double = 0
    var totalBlue: Double = 0
    var totalPercentage: Double = 0
    
    for colorItem in dominantColors {
        guard let rgb = colorItem.color, let percentage = colorItem.pixelPercentage else { continue }
        totalRed += Double(rgb.red) * percentage
        totalGreen += Double(rgb.green) * percentage
        totalBlue += Double(rgb.blue) * percentage
        totalPercentage += percentage
    }
    
    if totalPercentage > 0 {
        let avgRed = UInt8(totalRed / totalPercentage)
        let avgGreen = UInt8(totalGreen / totalPercentage)
        let avgBlue = UInt8(totalBlue / totalPercentage)
        print("平均RGB颜色：(\(avgRed), \(avgGreen), \(avgBlue))")
        // 转换为Hex格式
        let hexColor = String(format: "#%02X%02X%02X", avgRed, avgGreen, avgBlue)
        print("平均Hex颜色：\(hexColor)")
    }
}

方法2：直接使用背景/前景主色 #

启用图像属性检测后，response.imageProperties?.background和foreground会返回对应区域的主色，可直接调用：

if let bgColor = response.imageProperties?.background?.dominantColors?.first {
    print("背景主色CSS：\(bgColor.cssColor ?? "N/A")")
    print("背景主色Hex：\(bgColor.hexCode ?? "N/A")")
}
if let fgColor = response.imageProperties?.foreground?.dominantColors?.first {
    print("前景主色CSS：\(fgColor.cssColor ?? "N/A")")
    print("前景主色Hex：\(fgColor.hexCode ?? "N/A")")
}

注意事项 #

• 确保你的AWS IAM权限包含rekognition:DetectLabels操作。
• 图像格式需符合AWS Rekognition支持的类型（JPEG、PNG等）。

内容的提问来源于stack exchange，提问作者Gurmeet kaur