Image Classification Model #coreml

image_classification_model.swift

import CoreML
import Vision

func detectImage(image: CIImage) {
        guard let model = try? VNCoreMLModel(for: FlowerClassifier().model) else {
            fatalError("Could not load CoreML model")
        }
        
        let request = VNCoreMLRequest(model: model) { (request, err) in
            guard let results = request.results as? [VNClassificationObservation] else {
                fatalError("Could not retrieve result")
            }
            
            if let firstResult = results.first {
                self.navigationItem.title = firstResult.identifier
            }
            
        }
        
        let handler = VNImageRequestHandler(ciImage: image)
        do {
            try handler.perform([request])
        } catch {
            print("Could not perform the request \(error)")
        }
    }