Convert Image to CVPixelBuffer for Machine Learning Swift

by

You don’t need to do a bunch of image mangling yourself to use a Core ML model with an image — the new Vision framework can do that for you.

import Vision
import CoreML

let model = try VNCoreMLModel(for: MyCoreMLGeneratedModelClass().model)
let request = VNCoreMLRequest(model: model, completionHandler: myResultsMethod)
let handler = VNImageRequestHandler(url: myImageURL)
handler.perform([request])

func myResultsMethod(request: VNRequest, error: Error?) {
    guard let results = request.results as? [VNClassificationObservation]
        else { fatalError("huh") }
    for classification in results {
        print(classification.identifier, // the scene label
              classification.confidence)
    }

}

The WWDC17 session on Vision should have a bit more info — it’s tomorrow afternoon.

More Related Contents:

  1. How to get boolean value from switch state?
  2. Converting String to Int with Swift
  3. iOS 9 not opening Instagram app with URL SCHEME
  4. UITableView example for Swift
  5. Save string to the NSUserDefaults?
  6. Move textfield when keyboard appears swift
  7. Get the length of a String
  8. Add an element to an array in Swift
  9. How do you add an in-app purchase to an iOS application?
  10. How can I scan barcodes on iOS? [closed]
  11. Fade/dissolve when changing UIImageView’s image
  12. Subclass UIApplication with Swift
  13. Create space at the beginning of a UITextField
  14. Make a UIButton programmatically in Swift
  15. One timer per table view cell
  16. In Swift how to call method with parameters on GCD main thread?
  17. How can I create a UILabel with strikethrough text?
  18. How can I change the font of the back button for my navigation bar?
  19. Best way to parse URL string to get values for keys?
  20. How to get UIScrollView vertical direction in Swift?
  21. swift: issue in converting string to double
  22. Uncaught exception: This class is not key value coding-compliant [duplicate]
  23. How to change UIButton image in Swift
  24. Catch an exception for invalid user input in swift
  25. Show iPhone cut copy paste menu on UILabel
  26. Repeating local notification daily at a set time with swift
  27. Expected declaration error for making high score [duplicate]
  28. How to copy text to clipboard/pasteboard with Swift
  29. UIView’s border color in Interface builder doesn’t work?
  30. Detect backspace Event in UITextField

Leave a Comment