我的项目是使用 Vision 来识别视野中的某个矩形。我知道这个矩形将具有某种颜色。在 OpenCV 中,有一个过滤器可让您按颜色过滤像素,如此处所示。
我是 CoreImage 框架的新手,但我看不到任何可以实现此目标的过滤器。为此,是否有某种内置过滤器(不一定必须在 CoreImage 中),还是我需要学习如何编写自己的过滤器?
我的项目是使用 Vision 来识别视野中的某个矩形。我知道这个矩形将具有某种颜色。在 OpenCV 中,有一个过滤器可让您按颜色过滤像素,如此处所示。
我是 CoreImage 框架的新手,但我看不到任何可以实现此目标的过滤器。为此,是否有某种内置过滤器(不一定必须在 CoreImage 中),还是我需要学习如何编写自己的过滤器?
目标
我可以从给定链接下的信息中了解到,您想将红色区域过滤为白色,将所有其他区域过滤为黑色。
为此,可以使用 CIColorKernel。
CIColorKernel
基于 GPU 的图像处理例程,仅处理图像中的颜色信息,用于创建自定义核心图像过滤器。
见https://developer.apple.com/documentation/coreimage/cicolorkernel
截屏
这是一个背景颜色难看的屏幕截图,可以更好地查看图像覆盖的区域。
在上部区域显示输入图像。在下部区域中显示生成的图像。
资源
import UIKit
import CoreImage
class ViewController: UIViewController {
@IBOutlet weak var upperImageView: UIImageView!
@IBOutlet weak var lowerImageView: UIImageView!
let kernelString =
"""
kernel vec4 thresholdFilter(__sample textureColor) {
if (textureColor.r > 0.8 && textureColor.g < 0.1 && textureColor.b < 0.1) {
textureColor.rgb = vec3(1.0, 1.0, 1.0);
} else {
textureColor.rgb = vec3(0.0, 0.0, 0.0);
}
return textureColor;
}
"""
let kernel: CIColorKernel
required init?(coder aDecoder: NSCoder) {
self.kernel = CIColorKernel(source: kernelString)!
super.init(coder: aDecoder)
}
override func viewDidLoad() {
super.viewDidLoad()
self.upperImageView.image = UIImage(named: "example.png")
}
@IBAction func onFilter(_ sender: Any) {
guard let inputImage = upperImageView.image else { return }
guard let ciInputImage = CIImage(image: inputImage) else { return }
let ciImageFiltered = self.kernel.apply(
extent: ciInputImage.extent,
arguments: [ciInputImage])
let filteredImage = UIImage(ciImage: ciImageFiltered!)
self.lowerImageView.image = filteredImage
}
}
那是你要找的吗?