[SWIFT] [Core ML] Incorporez un modèle ML généré automatiquement dans l'application

Quel est cet article?

Il s'agit de la procédure pour introduire le modèle unique formé dans Create ML App dans l'application.

environnement

macOS 10.15.7 Xcode 12.1 Swift 5.3

procédure

Former le modèle

Cela se fait dans l'application Create ML. J'ai créé un modèle pour identifier Tomica (jouet minicar).

Incorporer dans le projet Xcode

Le fichier de modèle au format .mlmodel exporté depuis Create ML App peut être incorporé dans Xcode par simple glisser-déposer. En même temps, le code qui définit les classes du modèle ML est généré automatiquement.

Instance de modèle ML

Une partie du code au moment de l'annonce du framework CoreML était obsolète. Vous pouvez maintenant créer une instance avec le code suivant:

Créer une instance du modèle


let modelURL = Bundle.main.url(forResource: "MyTomicaClassifier", withExtension: "mlmodelc")!        
let tomicaClassifier = try! VNCoreMLModel(for: MLModel(contentsOf: modelURL))

Recommended Posts

[Core ML] Incorporez un modèle ML généré automatiquement dans l'application
Incorporer circleCI dans l'application CircleCI Rails
Intégrez Elasticsearch dans votre application Rails