web-dev-qa-db-ja.com

UIImageをCVPixelBufferに変換する方法

Appleの新しいCoreMLフレームワークには、CVPixelBufferを取る予測関数があります。 UIImageを分類するには、2つの間で変換を行う必要があります。

Appleエンジニアから得た変換コード：

1  // image has been defined earlier
2
3     var pixelbuffer: CVPixelBuffer? = nil
4   
5     CVPixelBufferCreate(kCFAllocatorDefault, Int(image.size.width), Int(image.size.height), kCVPixelFormatType_OneComponent8, nil, &pixelbuffer)
6     CVPixelBufferLockBaseAddress(pixelbuffer!, CVPixelBufferLockFlags(rawValue:0))
7   
8     let colorspace = CGColorSpaceCreateDeviceGray()
9     let bitmapContext = CGContext(data: CVPixelBufferGetBaseAddress(pixelbuffer!), width: Int(image.size.width), height: Int(image.size.height), bitsPerComponent: 8, bytesPerRow: CVPixelBufferGetBytesPerRow(pixelbuffer!), space: colorspace, bitmapInfo: 0)!
10  
11    bitmapContext.draw(image.cgImage!, in: CGRect(x: 0, y: 0, width: image.size.width, height: image.size.height))

このソリューションはSwiftにあり、グレースケール画像用です。画像のタイプに応じて行う必要がある変更は次のとおりです。

  • 5行目| kCVPixelFormatType_OneComponent8別のOSTypeへ（kCVPixelFormatType_32ARGBはRGB）
  • 8行目| colorSpaceから別のCGColorSpace（RGBの場合はCGColorSpaceCreateDeviceRGB）
  • 9行目| bitsPerComponentからメモリのピクセルあたりのビット数（RGBの場合は32）
  • 9行目| bitmapInfoをゼロ以外のCGBitmapInfoプロパティに（kCGBitmapByteOrderDefaultがデフォルトです）
swiftuiimagecvpixelbuffercoreml
17
Ryan

このチュートリアルをご覧ください https://www.hackingwithswift.com/whats-new-in-ios-11 、コードはSwift 4

func buffer(from image: UIImage) -> CVPixelBuffer? {
  let attrs = [kCVPixelBufferCGImageCompatibilityKey: kCFBooleanTrue, kCVPixelBufferCGBitmapContextCompatibilityKey: kCFBooleanTrue] as CFDictionary
  var pixelBuffer : CVPixelBuffer?
  let status = CVPixelBufferCreate(kCFAllocatorDefault, Int(image.size.width), Int(image.size.height), kCVPixelFormatType_32ARGB, attrs, &pixelBuffer)
  guard (status == kCVReturnSuccess) else {
    return nil
  }

  CVPixelBufferLockBaseAddress(pixelBuffer!, CVPixelBufferLockFlags(rawValue: 0))
  let pixelData = CVPixelBufferGetBaseAddress(pixelBuffer!)

  let rgbColorSpace = CGColorSpaceCreateDeviceRGB()
  let context = CGContext(data: pixelData, width: Int(image.size.width), height: Int(image.size.height), bitsPerComponent: 8, bytesPerRow: CVPixelBufferGetBytesPerRow(pixelBuffer!), space: rgbColorSpace, bitmapInfo: CGImageAlphaInfo.noneSkipFirst.rawValue)

  context?.translateBy(x: 0, y: image.size.height)
  context?.scaleBy(x: 1.0, y: -1.0)

  UIGraphicsPushContext(context!)
  image.draw(in: CGRect(x: 0, y: 0, width: image.size.width, height: image.size.height))
  UIGraphicsPopContext()
  CVPixelBufferUnlockBaseAddress(pixelBuffer!, CVPixelBufferLockFlags(rawValue: 0))

  return pixelBuffer
}
27
onmyway133