在iOS

时间:2017-02-15 15:47:55

标签: swift core-image

我想知道是否可以使用Core Image二值化图像(仅转换为黑白)?

我使用OpenCV和GPUImage制作它,但是如果可能的话,它更喜欢使用Apple Core Image

3 个答案:

答案 0 :(得分:3)

是。您至少有两个选项,CIPhotoEffectMono或一个小的自定义CIColorKernel。

<强> CIPhotoEffectMono:

func createMonoImage(image:UIImage) -> UIImage {
    let filter = CIFilter(name: "CIPhotoEffectMono")
    filter!.setValue(CIImage(image: image), forKey: "inputImage")
    let outputImage = filter!.outputImage
    let cgimg = ciCtx.createCGImage(outputImage!, from: (outputImage?.extent)!)
    return UIImage(cgImage: cgimg!)
}

请注意,我写的很快,你可能需要收紧零回报。

<强> CIColorKernel:

FadeToBW GLSL(0.0因子全彩色,1.0因子无颜色):

kernel vec4 fadeToBW(__sample s, float factor) {
    vec3 lum = vec3(0.299,0.587,0.114);
    vec3 bw = vec3(dot(s.rgb,lum));
    vec3 pixel = s.rgb + (bw - s.rgb) * factor;
    return vec4(pixel,s.a);
}

下面的代码将此作为名为FadeToBW.cikernel的文件打开。您也可以将此字符串直接发布到openKernelFile来电。

Swift代码:

func createMonoImage(image:UIImage, inputColorFade:NSNumber) -> UIImage {
    let ciKernel = CIColorKernel(string: openKernelFile("FadeToBW"))
    let extent = image.extent
    let arguments = [image, inputColorFade]        
    let outputImage = ciKernel.applyWithExtent(extent, arguments: arguments)
    let cgimg = ciCtx.createCGImage(outputImage!, from: (outputImage?.extent)!)
    return UIImage(cgImage: cgimg!)
}

再次,添加一些警卫等。

答案 1 :(得分:1)

您可以为此使用MetalPerformanceShaders。还有CIImageProcessingKernel。 https://developer.apple.com/documentation/coreimage/ciimageprocessorkernel

这是所需类的代码。

 class ThresholdImageProcessorKernel: CIImageProcessorKernel {
    static let device = MTLCreateSystemDefaultDevice()
    override class func process(with inputs: [CIImageProcessorInput]?, arguments: [String : Any]?, output: CIImageProcessorOutput) throws {
        guard
            let device = device,
            let commandBuffer = output.metalCommandBuffer,
            let input = inputs?.first,
            let sourceTexture = input.metalTexture,
            let destinationTexture = output.metalTexture,
            let thresholdValue = arguments?["thresholdValue"] as? Float else  {
                return
        }

        let threshold = MPSImageThresholdBinary(
            device: device,
            thresholdValue: thresholdValue,
            maximumValue: 1.0,
            linearGrayColorTransform: nil)

        threshold.encode(
            commandBuffer: commandBuffer,
            sourceTexture: sourceTexture,
            destinationTexture: destinationTexture)
    }
}

这是您可以使用的方式:

    let context = CIContext(options: nil)

            if let binaryCIImage = try? ThresholdImageProcessorKernel.apply(
                withExtent: croppedCIImage.extent,
                inputs: [croppedCIImage],
                arguments: ["thresholdValue": Float(0.2)]) {
                if let cgImage = context.createCGImage(binaryCIImage, from: binary.extent) {
                    DispatchQueue.main.async {
                        let resultingImage = UIImage(cgImage: cgImage)
                        if resultingImage.size.width > 100 {
                            print("Received an image \(resultingImage.size)")
                        }
                    }
                }
            }

答案 2 :(得分:1)

通过使用 CIPhotoEffectMono 或等效方法将其转换为灰度,然后使用具有高得离谱的 inputContrast 数字(我使用 10000)的 CIColorControls,我取得了成功。这有效地使它变成黑白的,从而二值化。对于那些不想弄乱自定义内核的人很有用。

此外,您还可以使用像 Apple 的“Chroma Key”过滤器这样的示例,它使用 Hue 进行过滤,但不是查看 Hue,您只需给出对数据进行二值化的规则(即:何时将 RGB 全部设置为 1.0,何时将设置为 0.0)。

https://developer.apple.com/documentation/coreimage/applying_a_chroma_key_effect