swift - 如何在 Xcode 10 中创建 MLImageClassifier

Question

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

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

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

到现在为止我有这个：

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

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

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

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