vision.swift

extension LiveCameraViewController: LiveCameraDelegate {
    
    private let ageModel = try! VNCoreMLModel(for: AgeNet().model)
    private let genderModel = try! VNCoreMLModel(for: GenderNet().model)
    
    func createAgeRequest() -> VNCoreMLRequest {
        return VNCoreMLRequest(model: ageModel, completionHandler: onClassification)
    }
    
    func createGenderRequest() -> VNCoreMLRequest {
        return VNCoreMLRequest(model: genderModel, completionHandler: onClassification)
    }
    
    func onClassification(request: VNRequest, error: Error?) {
        guard
            let results = request.results as? [VNClassificationObservation],
            let result = results.first
        else {
            return
        }
        DispatchQueue.main.async { [weak self] in
            if results.count == 2 { // Gender
                self?.updateGender(label: result.identifier, confidence: result.confidence)
            } else { // Age
                self?.updateAge(label: result.identifier, confidence: result.confidence)
            }
        }
    }
    
    func didUpdateBuffer(_ buffer: CMSampleBuffer, on camera: LiveCamera) {
        let handler = VNImageRequestHandler(cvPixelBuffer: CMSampleBufferGetImageBuffer(buffer)!, options: [:])
        try? handler.perform([createAgeRequest(), createGenderRequest()])
    }
}