远距离颜色检测(Swift)

Color detection at a distance (Swift)

有人可以在远处用颜色检测为我指明正确的方向吗?我使用了下面的代码,如果物体或兴趣点距离小于 10 英尺,它会正确地抓取图像的 RBG 值。当物体在远处时,代码 returns 错误的值。我想拍摄距离大于 10 英尺的物体的照片并检测该图像的颜色。

 //On the top of your swift 
  extension UIImage {
  func getPixelColor(pos: CGPoint) -> UIColor {

      let pixelData = CGDataProviderCopyData(CGImageGetDataProvider(self.CGImage))
      let data: UnsafePointer<UInt8> = CFDataGetBytePtr(pixelData)

      let pixelInfo: Int = ((Int(self.size.width) * Int(pos.y)) + Int(pos.x)) * 4

      let r = CGFloat(data[pixelInfo]) / CGFloat(255.0)
      let g = CGFloat(data[pixelInfo+1]) / CGFloat(255.0)
      let b = CGFloat(data[pixelInfo+2]) / CGFloat(255.0)
      let a = CGFloat(data[pixelInfo+3]) / CGFloat(255.0)

      return UIColor(red: r, green: g, blue: b, alpha: a)
  }  
  }

我是一名摄影师,您正在尝试做的事情与在 post 处理中设置白平衡或在 PS 中使用颜色选择器非常相似。

数码相机的像素无法一次捕获全部光谱。他们有 RGB 像素的三元组。捕获的信息被插值,这会产生非常糟糕的结果。将夜间拍摄的图像设置为 post 的白平衡几乎是不可能的。

插值错误的原因:

  • 像素大于场景中最小的可辨别对象。 (摩尔纹)
  • 数字增益增加色差的低光情况。 (颜色噪声伪影)
  • 图像已转换为低质量的 jpg,但有很多边缘。 (jpg 工件)

如果是低画质的jpg,求个更好的源img。

修复

要获得更准确的读数,您只需模糊图像即可。 可接受的最小模糊是 3 个像素,因为这会撤消一些插值。更大的模糊可能更好。

由于模糊成本很高,因此最好将图像裁剪成模糊半径的倍数。您不能进行精确拟合,因为它也会模糊边缘,并且超出边缘图像是黑色的。这会影响你的阅读。

如果您也对模糊半径施加上限可能是最好的。


获取具有大小的东西的中心的快捷方式。

extension CGSize {

    var center : CGPoint {
        get {
            return CGPoint(x: width / 2, y: height / 2)
        }
    }
}

UIImage 东西

extension UIImage {

    func blur(radius: CGFloat) -> UIImage? {
        // extensions of UImage don't know what a CIImage is...
        typealias CIImage = CoreImage.CIImage

        // blur of your choice
        guard let blurFilter = CIFilter(name: "CIBoxBlur") else {
            return nil
        }

        blurFilter.setValue(CIImage(image: self), forKey: kCIInputImageKey)
        blurFilter.setValue(radius, forKey: kCIInputRadiusKey)

        let ciContext  = CIContext(options: nil)

        guard let result = blurFilter.valueForKey(kCIOutputImageKey) as? CIImage else {
            return nil
        }

        let blurRect = CGRect(x: -radius, y: -radius, width: self.size.width + (radius * 2), height: self.size.height + (radius * 2))

        let cgImage = ciContext.createCGImage(result, fromRect: blurRect)

        return UIImage(CGImage: cgImage)

    }

    func crop(cropRect : CGRect) -> UIImage? {

        guard let imgRef = CGImageCreateWithImageInRect(self.CGImage, cropRect) else {
            return nil
        }
        return UIImage(CGImage: imgRef)

    }

    func getPixelColor(atPoint point: CGPoint, radius:CGFloat) -> UIColor? {

        var pos = point
        var image = self

        // if the radius is too small -> skip
        if radius > 1 {

            let cropRect = CGRect(x: point.x - (radius * 4), y: point.y - (radius * 4), width: radius * 8, height: radius * 8)
            guard let cropImg = self.crop(cropRect) else {
                return nil
            }

            guard let blurImg = cropImg.blur(radius) else {
                return nil
            }

            pos = blurImg.size.center
            image = blurImg

        }

        let pixelData = CGDataProviderCopyData(CGImageGetDataProvider(image.CGImage))
        let data: UnsafePointer<UInt8> = CFDataGetBytePtr(pixelData)

        let pixelInfo: Int = ((Int(image.size.width) * Int(pos.y)) + Int(pos.x)) * 4

        let r = CGFloat(data[pixelInfo]) / CGFloat(255.0)
        let g = CGFloat(data[pixelInfo+1]) / CGFloat(255.0)
        let b = CGFloat(data[pixelInfo+2]) / CGFloat(255.0)
        let a = CGFloat(data[pixelInfo+3]) / CGFloat(255.0)

        return UIColor(red: r, green: g, blue: b, alpha: a)
    }  
}

旁注:

您的问题可能不是颜色抓取功能,而是您如何设置点。如果您是通过触摸操作,并且对象在屏幕上更远并因此更小,则您可能设置得不够准确。

读取 UIImage 的平均颜色 ==> https://www.hackingwithswift.com/example-code/media/how-to-read-the-average-color-of-a-uiimage-using-ciareaaverage

extension UIImage {
    var averageColor: UIColor? {
    guard let inputImage = CIImage(image: self) else { return nil }
    let extentVector = CIVector(x: inputImage.extent.origin.x, y: inputImage.extent.origin.y, z: inputImage.extent.size.width, w: inputImage.extent.size.height)

    guard let filter = CIFilter(name: "CIAreaAverage", parameters: [kCIInputImageKey: inputImage, kCIInputExtentKey: extentVector]) else { return nil }
    guard let outputImage = filter.outputImage else { return nil }

    var bitmap = [UInt8](repeating: 0, count: 4)
    let context = CIContext(options: [.workingColorSpace: kCFNull])
    context.render(outputImage, toBitmap: &bitmap, rowBytes: 4, bounds: CGRect(x: 0, y: 0, width: 1, height: 1), format: .RGBA8, colorSpace: nil)

    return UIColor(red: CGFloat(bitmap[0]) / 255, green: CGFloat(bitmap[1]) / 255, blue: CGFloat(bitmap[2]) / 255, alpha: CGFloat(bitmap[3]) / 255)
    }
}