Advertisement
Guest User

Untitled

a guest
Jun 28th, 2017
56
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.08 KB | None | 0 0
  1. extension LiveCameraViewController: LiveCameraDelegate {
  2.  
  3. private var ageRequest: VNCoreMLRequest = {
  4. let model = try! VNCoreMLModel(for: AgeNet().model)
  5. return VNCoreMLRequest(model: model) { request, error in
  6. if let observation = request.results?.first as? VNClassificationObservation {
  7. // do something
  8. }
  9. }
  10. }()
  11.  
  12. private var genderRequest: VNCoreMLRequest = {
  13. let model = try! VNCoreMLModel(for: GenderNet().model)
  14. return VNCoreMLRequest(model: model) { request, error in
  15. if let observation = request.results?.first as? VNClassificationObservation {
  16. // do something
  17. }
  18. }
  19. }()
  20.  
  21. func didUpdateBuffer(_ buffer: CMSampleBuffer, on camera: LiveCamera) {
  22. let image = UIImage(buffer: CMSampleBufferGetImageBuffer(buffer)!)?
  23. .rescaled(width: inputImageScale, height: inputImageScale)
  24. .cgImage
  25. let handler = VNImageRequestHandler(cgImage: image!)
  26. try? handler.perform([ageRequest, genderRequest])
  27. }
  28. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement