Skip to content

Instantly share code, notes, and snippets.

@anupamchugh
Created August 21, 2020 15:38
Show Gist options
  • Save anupamchugh/c2b1f5e71a8cbbdd583873e39f8c4f56 to your computer and use it in GitHub Desktop.
Save anupamchugh/c2b1f5e71a8cbbdd583873e39f8c4f56 to your computer and use it in GitHub Desktop.
guard let model = try? VNCoreMLModel(for: StyleBlue.init(configuration: config).model) else { return }
let request = VNCoreMLRequest(model: model) { (finishedRequest, error) in
guard let results = finishedRequest.results as? [VNPixelBufferObservation] else { return }
guard let observation = results.first else { return }
DispatchQueue.main.async(execute: {
self.imageView.image = UIImage(pixelBuffer: observation.pixelBuffer)
})
}
guard let pixelBuffer: CVPixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else { return }
try? VNImageRequestHandler(cvPixelBuffer: pixelBuffer, options: [:]).perform([request])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment