如何從 CVPixelBuffer 創建 CGImage？

Question

我有一個簡單的 function 從CVPixelBuffer CGImage

if let buffer = (results as? [VNPixelBufferObservation])?.first {
    let pixelBuffer = buffer.pixelBuffer
    let ciImage = CIImage(cvPixelBuffer: pixelBuffer)
    let resultImage = UIImage(ciImage: ciImage)
    return resultImage.cgImage //returns nil
}

但它返回零。 為什么？

Answer 1

output 為 nil，因為您正在使用 CIImage 而不是 CGImage 創建 UIImage 實例。 這是創建CGImage的一種方法：

func createCGImage(from pixelBuffer: CVPixelBuffer) -> CGImage? {
   let ciContext = CIContext()
   let ciImage = CIImage(cvImageBuffer: pixelBuffer)
   return context.createCGImage(ciImage, from: ciImage.extent)
}

Answer 2

試試這個

import VideoToolbox
func createImage(from pixelBuffer: CVPixelBuffer) -> CGImage? {
    var cgImage: CGImage?
    VTCreateCGImageFromCVPixelBuffer(pixelBuffer, options: nil, imageOut: &cgImage)
    return cgImage
}