在 objective-C 中获取 coreml 运行

Question

我在创建一个在 Objective-C 中使用 Apple 的 CoreML 的非常简单的示例时遇到了问题。我已经使用 python 创建了一个模型文件，现在工作起来相当简单：

coreml_model_svm = coremltools.models.MLModel("svm.mlmodel")
test_x = [1.0 for x in range(160)]
predictions_coreml_svm = coreml_model_svm.predict({"input":test_x})

我想在Objective-C中重现以上三行。我知道我的数据必须是 MLMultiArray 并且模型需要加载到 MLModel。我一直在尝试查找有关语法的一些信息，但似乎我不明白文档是如何工作的，所有示例都在 Swift 中。到目前为止，这是我的代码。注释 MLMultiArray 会导致初始化 MLModel 时出现未捕获的异常。当不评论 MLMultiArray 我得到 no known class method for selector 'initWithShape:dataType:error'.

#import <Foundation/Foundation.h>
#import <CoreML/CoreML.h>

//clang -framework Foundation coremltest.m -o coremltest
int main (int argc, const char * argv[])
{
        NSLog(@"start");

        NSArray * shape = [[NSArray alloc]  init];
        MLMultiArrayDataType dataType = MLMultiArrayDataTypeDouble;
        NSError * error = nil;

        MLMultiArray * input =  [MLMultiArray initWithShape:(NSArray*) shape
                             dataType:(MLMultiArrayDataType ) dataType
                        error:(NSError **) error];

        MLModel * mymodel = [[MLModel init] initWithContentsOfFile:@"svm.mlmodel"];

        return 0;
}

如有任何帮助，我将不胜感激。

Answer 1

首先你需要导入你的模型class，在你的情况下是svm（最好是Svm以大写字母开头）：

#import "svm.h"

此 class 定义包含输入和输出以及方法定义所需的所有信息。当您选择左侧的 mlmodel 文件时，您可以通过单击 class 名称旁边的小箭头在自动生成的 class 描述中找到此规范。在此class描述中

在您的例子中，输入是一个 MLMultiArray，其中包含 160 个 Double 元素作为向量。所以首先用 shape array

定义维度

NSArray *shape = @[@1, @160];

然后定义 MLMultiArray，它将成为预测过程的 svmModelInput（再次由 XCode 自动定义）：

    MLMultiArrayDataType dataType = MLMultiArrayDataTypeDouble;
    NSError *error = nil;

    MLMultiArray *theMultiArray =  [[MLMultiArray alloc] initWithShape:(NSArray*)shape
                                          dataType:(MLMultiArrayDataType)dataType
                                             error:&error] ;

    for (int i = 0; i < 160; i++) {
         [theMultiArray setObject:[NSNumber numberWithDouble:1.0] atIndexedSubscript:(NSInteger)i];
    }

二手

initWithShape

是 Apple 的一种 MLMultiArray 方法。我用“1”填充数组只是为了测试，但你当然必须稍后用你的真实输入替换它。

无需获取模型，只需实例化 svm 然后运行

predictionFromInput:

方法来自XCode再次构建的class：

        svm *mySvm = [[svm alloc] init];

        svmOutput * svmModelOutput = [(svm *)mySvm predictionFromInput:theMultiArray error:&error];
        NSLog(@"SVM Model output = %lld -- %@", svmModelOutput.classLabel, svmModelOutput.classProbability );

           if (!error)
           {
               NSLog(@"svm finished without error");
           }
           else
           {
               NSLog(@"Error: %@", error.localizedDescription);
           }

当您打印出 svmModelOutput（由 XCode 为您创建）时，您可以检查预测的 classLabel 作为整数和所有标签，如下所示：

2017-12-04 07:32:45.765015+0100 CoreML_test[2634:877638] SVM Model output = 2 -- {
    11 = "0.002656571278812773";
    3 = "0.2121030282896462";
    10 = "0.004570897664662783";
    2 = "0.5825387375626612";
    9 = "0.02911120023388797";
    4 = "0.1690195649703292";
}

在 objective-C 中获取 coreml 运行

getting coreml running in objective-C

objective-c

coreml