在 swift 中使用 CIImage 滤镜控制黑白着色

Question

我的代码将彩色 NSImage 转换为灰度。然后它将灰度转换为纯黑白图像。有没有办法创建一个可以指定如何将灰色像素转换为黑色/白色的过滤器？例如 - 1.如果像素值> 100，将其变为黑色。 2.否则将像素变成白色。

我可以使用哪些过滤器设置来实现此目的？

更新了 CIKernel 自定义过滤器：

func applyCustomCIFilter( path: String )->NSImage{
    let url = URL(fileURLWithPath: path)
    let origImage = CIImage(contentsOf: url)!

    // Convert image into grayscale
    let grayImage = CIFilter(name: "CIPhotoEffectNoir", parameters: [kCIInputImageKey: origImage])?.outputImage
    print("convert into black and white")

    // custom filter to convert light gray to white and dark gray to black
    let replaceGrayKernel = CIColorKernel( source:
        "kernel vec4 replaceGrayWithBlackOrWhite(sampler grayImage) {" +
            "if( sample(grayImage, samplerCoord(grayImage)).rgb > 0.7  ){" +
            "return vec4(0.0,0.0,0.0,1.0);" +
            "}" +
            "else" +
            "{" +
            "return vec4(1.0,1.0,1.0,1.0);" +
            "}" +
        "}"
    )

    //Apply custom filter to grayscale image. 
    //ERROR: blackAndWhiteImage is nil causing runtime failure
    let blackAndWhiteImage = replaceGrayKernel?.apply(extent: (grayImage!.extent), arguments: [grayImage as Any])

    //Convert CIImage to NSImage
    let rep = NSCIImageRep(ciImage: blackAndWhiteImage!)
    let nsImage = NSImage(size: rep.size)
    nsImage.addRepresentation(rep)

    return nsImage;
}

Answer 1

您的内核代码有 2 个错误。这是正确的代码：

let kernel = CIColorKernel( source:
    "kernel vec4 replaceGrayWithBlackOrWhite(__sample s) {" +
        "if (s.r > 0.25 && s.g > 0.25 && s.b > 0.25) {" +
        "    return vec4(0.0,0.0,0.0,1.0);" +
        "} else {" +
        "    return vec4(1.0,1.0,1.0,1.0);" +
        "}" +
    "}"
)

从 sampler 到 __ sample 的变化是因为 CIColorKernel，它针对一次处理一个像素进行了优化，只是传递到内核中那个像素。因此，也不需要调用 sample(samplerCoord())。 CIWarpKernel 和 CIKernel 使用 sampler，因为您将 RIO（感兴趣区域）发送到内核，这些内核可以访问周围的像素……想想模糊效果。
第二个变化是 if 语句。 if (s.rgb > 0.7) 将 vec3（或三个浮点数）与单个浮点数进行比较。一旦我纠正了这些值，我就不得不玩弄这些值，并认为 (a) 使用 AND (&&) 代替或 OR (||) 以及 (b) 将阈值降低到 0.25 使得更接近black/white 图片。自己动手玩一下，看看你想要什么。

我创建了一个使用此内核的 small Swift 5 project（使用 hard-coded 图像）。

其中没有注释，它包含各种扩展（和 GLKView 的子类），因为我从一个生产项目中收集了一些东西。除了关注您的内核代码问题外，还有一个关于代码的警告词 - 它包含几个 force-unwraps 应该为 "production-ready" 代码删除。

在 swift 中使用 CIImage 滤镜控制黑白着色

Control black and white coloring using CIImage filter in swift

swift

filter

nsimage

ciimage