Skip to content

Instantly share code, notes, and snippets.

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 emirerdo/790f56bc545335c75d0334c9fcb8664d to your computer and use it in GitHub Desktop.
Save emirerdo/790f56bc545335c75d0334c9fcb8664d to your computer and use it in GitHub Desktop.
func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
guard let pixelBuffer: CVPixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else {
return
}
guard let model = try? VNCoreMLModel(for: MobileNet().model) else {
return
}
let request = VNCoreMLRequest(model: model) { (req, err) in
guard let results = req.results as? [VNClassificationObservation] else {
return
}
guard let firstResult = results.first else {
return
}
print("Detected item: \(firstResult.identifier), Confidence: \(firstResult.confidence)")
}
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