手把手教你如何快速解析.mlmodel文件

把训练好的模型拖入工程

把.mlmodel文件拖入到 iOS程序后，点击.mlmodel文件他会显示这个界面

• Model Class
  • 这个是 Xcode 根据模型编译生成的接口类，应用程序需要 #import “imageClassifier.h”。点最右边那个箭头，就会打开这个头文件。
• Model Evaluation Parameters
  • 描述输入和输出，CoreML 接的机器学习模型是一个黑箱，我们只管从进料孔放原料进去，过一会儿，加工好的结果就会从输出口送出来了。每个模型解决的问题、处理的方法不同，需要的输入和产生的输出也各不相同。
  • 输入为images。
    • images的type为Image<RGB,224,224> 表示接收一个图片，格式得是 RGB 的，大小得是 224x224 像素的。
  • 输出的结果是两个参数，一个是 sceneLabel 表示图片中最有可能是什么场合，另一个是 sceneLabelProbs 是所有场景分别的可能性。

.mlmodel文件解析

第一部分input

/// Model Prediction Input Type
API_AVAILABLE(macos(10.13), ios(11.0), watchos(4.0), tvos(11.0)) __attr