Created
February 29, 2020 04:21
-
-
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
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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