UIImage को एक CVPixelBuffer

Question

में कनवर्ट करने के लिए कैसे करें Apple के नए CoreML ढांचे में भविष्यवाणी फ़ंक्शन है जो CVPixelBuffer लेता है। UIImage वर्गीकृत करने के लिए दोनों के बीच एक रूपांतरण किया जाना चाहिए।

रूपांतरण कोड मैं एक एप्पल इंजीनियर से मिला:

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)) 

यह समाधान तेजी में है और एक ग्रेस्केल छवि के लिए है। छवि के प्रकार के आधार पर किए गए परिवर्तन हैं:

• रेखा 5 | kCVPixelFormatType_OneComponent8 अन्य OSType (kCVPixelFormatType_32ARGB आरजीबी के लिए)
• लाइन 8 | colorSpace अन्य CGColorSpace (CGColorSpaceCreateDeviceRGB आरजीबी के लिए)
• लाइन 9 | bitsPerComponent बिट्स प्रति पिक्सेल की बिट्स (आरजीबी के लिए 32)
• लाइन 9 | bitmapInfo एक अशून्य CGBitmapInfo संपत्ति के लिए,

Answer 1

आप इस ट्यूटोरियल https://www.hackingwithswift.com/whats-new-in-ios-11 पर एक नज़र ले जा सकते हैं (kCGBitmapByteOrderDefault डिफ़ॉल्ट है) कोड स्विफ्ट 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 
}