Swift中基于CIFilter实现签名扫描器并移除白色背景的技术求助

你的核心问题是没有正确构建区分白色背景和蓝色签名的掩码逻辑，原代码的colorMatrix设置完全屏蔽了RGB通道，导致输出黑色图像。以下是可行的实现方案：

实现思路 #

1. 先通过颜色过滤提取蓝色签名的区域，生成对应的alpha掩码
2. 将原图像与掩码结合，将白色背景区域的alpha设为0（透明），同时保留蓝色签名的色彩

完整代码实现 #

func filterImage() {
    guard let inputImage = inputImage, let beginImage = CIImage(image: inputImage) else { return }
    let context = CIContext()
    
    // 步骤1: 增强蓝色对比度，让签名与背景更易区分
    let contrastFilter = CIFilter.colorControls()
    contrastFilter.inputImage = beginImage
    contrastFilter.contrast = 2.0 // 提高对比度，强化蓝色与白色的差异
    contrastFilter.saturation = 1.5 // 提升饱和度，让蓝色更鲜艳
    guard let contrastedImage = contrastFilter.outputImage else { return }
    
    // 步骤2: 创建掩码 - 蓝色区域设为不透明，白色区域设为透明
    let maskFilter = CIFilter.colorMatrix()
    maskFilter.inputImage = contrastedImage
    
    // 颜色矩阵逻辑：提取蓝色通道的差异，生成alpha掩码
    // R/G通道设为0，只保留蓝色通道的反向值作为alpha（白色的B通道值高，alpha趋近于0；蓝色B通道值相对低，alpha趋近于1）
    maskFilter.setValue(CIVector(x: 0, y: 0, z: 0, w: 0), forKey: "inputRVector")
    maskFilter.setValue(CIVector(x: 0, y: 0, z: 0, w: 0), forKey: "inputGVector")
    maskFilter.setValue(CIVector(x: 0, y: 0, z: 0, w: 0), forKey: "inputBVector")
    // alpha通道由 (255 - B值) 计算，白色B值接近255，alpha为0；蓝色B值低，alpha为1
    maskFilter.setValue(CIVector(x: 0, y: 0, z: -1, w: 1), forKey: "inputAVector")
    
    guard let maskImage = maskFilter.outputImage else { return }
    
    // 步骤3: 将原图像与掩码合成，实现透明背景
    let compositeFilter = CIFilter.blendWithMask()
    compositeFilter.inputImage = contrastedImage
    compositeFilter.maskImage = maskImage
    guard let maskedImage = compositeFilter.outputImage else { return }
    
    // 转换为UIImage输出
    if let cgimg = context.createCGImage(maskedImage, from: maskedImage.extent) {
        let uiimage = UIImage(cgImage: cgimg)
        outputImageMain = uiimage
    }
}

关键细节说明 #

• 对比度与饱和度调整：先强化蓝色签名和白色背景的差异，能大幅提升后续掩码的准确性
• 颜色矩阵构建：通过反向使用蓝色通道值生成alpha掩码，白色背景的蓝色通道值高，对应alpha为0（透明）；蓝色签名的蓝色通道值相对低，对应alpha为1（不透明）
• blendWithMask滤镜：直接利用掩码控制原图像的透明度，避免手动处理像素的繁琐

如果拍摄的照片存在阴影或偏色，可以在步骤1前添加CIPhotoEffectNoir或CIColorBalance滤镜进一步优化输入图像的纯净度。

内容的提问来源于stack exchange，提问作者Rin Star