如何在 Xcode 10 中创建 MLImageClassifier

Question

我正在尝试在 Xcode 10.1 Playground 中使用 Create ML 构建机器学习 - Image Recognition，但我在将数据放入模型时遇到了一些问题。

我有一个包含编号为 1 到 1336 的图像的文件夹和一个包含 2 列（图像名称和图像分类）的 .csv 文件。

我不知道如何将它放入模型中。

我到现在都有这个：

import Cocoa
import CreateML

let data = try MLDataTable(contentsOf: URL(fileURLWithPath: "/Users/x/Desktop/CoreML/project/file.csv"))

let(trainingData, testingData) = data.randomSplit(by: 0.8, seed: 1)

let Classifier = try MLImageClassifier *need help here*

let evaluationMetrics = sentimentClassifier.evaluation(on: testingData)

let evaluationAccuracy = (1 - evaluationMetrics.classificationError) * 100

let metaData = MLModelMetadata(author: "x", shortDescription: "Model", version: "1.0")

try classifier.write(to: URL(fileURLWithPath: "/Users/x/Desktop/CoreML/project/XClassifier.mlmodel"))

Answer 1

我认为无法通过.csv 或任何其他单独的文件将标签提供给MLImageClassifier。您只有两个选择：使用文件名作为标签或使用目录作为标签（在您的许多图像的情况下可能更可取）：

let model = try MLImageClassifier(trainingData: .labeledDirectories(at: trainingDir))
let evaluation = model.evaluation(on: .labeledDirectories(at: testingDir))

您需要将图像放入.csv 文件中以标签命名的子目录中。