【Core ML】自動生成されるMLモデルをアプリに組み込む

この記事はなに？

Create ML Appでトレーニングした独自のモデルを、アプリに導入するまでの手続きです。

環境

macOS 10.15.7
Xcode 12.1
Swift 5.3

手順

モデルをトレーニングする

この作業は、Create ML Appで行います。
トミカ（ミニカーの玩具）を識別するモデルを作成しました。

Xcodeプロジェクトに組み込む

Create ML Appから書き出した.mlmodel形式のモデルファイルは、ドラッグ&ドロップするだけでXcodeに組み込むことができます。
同時に、MLモデルのクラスを定義したコードが自動的に生成されます。

MLモデルのインスタンス

CoreMLフレームワーク発表当時のコードは一部、非推奨になっています。
現在は以下のコードで、インスタンスを作成できます。

モデルのインスタンスを作成する

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