swift - 平滑色度键的边缘 -CoreImage

Question

我正在使用以下代码从图像中删除绿色背景。但是图像的边缘有绿色色调并且一些像素已损坏。我如何平滑它并使剪切完美。

 func chromaKeyFilter(fromHue: CGFloat, toHue: CGFloat) -> CIFilter?
    {
        // 1
        let size = 64
        var cubeRGB = [Float]()

        // 2
        for z in 0 ..< size {
            let blue = CGFloat(z) / CGFloat(size-1)
            for y in 0 ..< size {
                let green = CGFloat(y) / CGFloat(size-1)
                for x in 0 ..< size {
                    let red = CGFloat(x) / CGFloat(size-1)

                    // 3
                    let hue = getHue(red: red, green: green, blue: blue)
                    let alpha: CGFloat = (hue >= fromHue && hue <= toHue) ? 0: 1

                    // 4
                    cubeRGB.append(Float(red * alpha))
                    cubeRGB.append(Float(green * alpha))
                    cubeRGB.append(Float(blue * alpha))
                    cubeRGB.append(Float(alpha))
                }
            }
        }



  @IBAction func clicked(_ sender: Any) {
          let a = URL(fileURLWithPath:"green.png")
          let b = URL(fileURLWithPath:"back.jpg")

        let image1 = CIImage(contentsOf: a)
        let image2 = CIImage(contentsOf: b)





        let chromaCIFilter = self.chromaKeyFilter(fromHue: 0.3, toHue: 0.4)
        chromaCIFilter?.setValue(image1, forKey: kCIInputImageKey)
        let sourceCIImageWithoutBackground = chromaCIFilter?.outputImage

        /*let compositor = CIFilter(name:"CISourceOverCompositing")
        compositor?.setValue(sourceCIImageWithoutBackground, forKey: kCIInputImageKey)
        compositor?.setValue(image2, forKey: kCIInputBackgroundImageKey)
        let compositedCIImage = compositor?.outputImage*/

        var rep: NSCIImageRep = NSCIImageRep(ciImage: sourceCIImageWithoutBackground!)
        var nsImage: NSImage = NSImage(size: rep.size)
        nsImage.addRepresentation(rep)

        let url = URL(fileURLWithPath:"file.png")

        nsImage.pngWrite(to: url)
        super.viewDidLoad()
    }

输入:

输出:

更新:

更新 2:

更新 3:

Answer 1

用于色度键控的专业工具通常包括所谓的溢出抑制器。溢出抑制器会发现包含少量色度键颜色的像素，并将颜色向相反方向移动。所以绿色像素会向洋红色移动。这减少了您经常在键控镜头周围看到的绿色边缘。

您称为损坏的像素只是其中具有一定程度的色度颜色并且正在被您的键控器功能拾取的像素。与其选择硬性的 0 或 1，不如考虑一个基于像素颜色返回 0 到 1 之间的值的函数。例如，您可以找到当前像素色调与 fromHue 和 toHue 的角距离，并可能执行如下操作:

// Get the distance from the edges of the range, and convert to be between 0 and 1
var distance: CGFloat
if (fromHue <= hue) && (hue <= toHue) {
    distance = min(abs(hue - fromHue), abs(hue - toHue)) / ((toHue - fromHue) / 2.0)
} else {
    distance = 0.0
}
distance = 1.0 - distance
let alpha = sin(.pi * distance - .pi / 2.0) * 0.5 + 0.5

这将为您提供从范围边缘到范围中心的平滑变化。 (请注意，我没有处理色调以 360° 环绕的事实。这是您必须处理的事情。)衰减图如下所示:

您可以做的另一件事是将键控限制为仅影响饱和度高于某个阈值且值高于某个阈值的像素。对于非常暗和/或不饱和的颜色，您可能不想将其抠出。例如，我认为这有助于解决您在模特夹克上看到的问题。