远距离颜色检测(Swift)

时间:2015-11-15 17:27:06

标签: ios swift colors uiimage rgb

有人能用正确的方向指出我在远处进行颜色检测吗?我已经使用了下面的代码,如果对象或兴趣点距离不到10英尺,它会正确地抓取图像的RBG值。当对象处于距离时,代码返回错误的值。我想拍摄距离大于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)
  }  
  }

2 个答案:

答案 0 :(得分:4)

我是一名摄影师,你要做的就是在后期处理中设置白平衡或在PS中使用颜色选择器非常相似。

数码相机没有能够同时捕捉全部光线的像素。它们具有RGB三像素像素。捕获的信息是内插的,这可能会产生非常糟糕的结果。在夜间拍摄的图像上设置白平衡几乎是不可能的。

插值错误的原因:

  • 像素大于场景中最小的可辨别对象。 (莫尔文物)
  • 低光照情况,数字增益会增加色差。 (色噪声文物)
  • 图像转换为低质量的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)
    }  
}

旁注:

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

答案 1 :(得分:0)

读取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)
    }
}