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°环绕的事实。这是您必须要处理的事情。）衰减图如下所示：

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

Answer 2

我的（实时）抠像器的工作方式如下（user1118321描述了增强功能），并使用其分析器，我很快注意到这很可能不是真正的绿屏图像。这是许多伪造的产品之一，其中的绿色屏幕似乎已被饱和的单色绿色代替。尽管这看起来不错，但它会引入伪影，其中抠像的对象（带有最初使用的绿色的边缘）与单色绿色相遇。通过查看直方图，可以看到使用了一个绿色。真正的绿色屏幕始终具有（可见）绿色阴影。我能够得到一个不错的钥匙，但不得不手动调整一些设置。使用NLE，您可能会获得更好的结果，但是它们也会变得更加复杂。因此，要解决您的问题，您的代码可能像现在一样工作（更新＃3），您只需要使用适当的真实绿色屏幕图像即可。

平滑色度键-CoreImage的边缘

2 个答案: