如何在Xcode 10中创建MLImageClassifier

Question

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

我有一个文件夹，图像编号从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

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

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

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