Skip to content

Instantly share code, notes, and snippets.

@digitallysavvy
Created February 29, 2020 04:21
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save digitallysavvy/e30e1c4cc3a569e5635864c689b5f918 to your computer and use it in GitHub Desktop.
Save digitallysavvy/e30e1c4cc3a569e5635864c689b5f918 to your computer and use it in GitHub Desktop.
Example snippet of how to run a CoreML ImageRequest on an ARKit scene's current frame
func runCoreML() {
// Get Camera Image as RGB
guard let sceneView = self.sceneView else { return }
guard let currentFrame = sceneView.session.currentFrame else { return }
let pixbuff : CVPixelBuffer = currentFrame.capturedImage
let ciImage = CIImage(cvPixelBuffer: pixbuff)
// Prepare CoreML/Vision Request
let imageRequestHandler = VNImageRequestHandler(ciImage: ciImage, options: [:])
// Run Image Request
do {
try imageRequestHandler.perform(self.visionRequests)
} catch {
print(error)
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment