如何在CoreML中访问模型的描述？

编程问答 2022-06-06

问题描述

我有一个CoreML模型，并且已经使用coremltools向该模型添加了信息：

model.author = 'vincent Garcia'
model.license = 'BSD'
model.short_description = 'The model is doing something.'

是否可以通过Swift访问此信息？

在Apple的文档中，其内容为：

检查模型的元数据和MLFeatureDescription实例 通过modelDescription。

我尝试过：

let model = try! MyModel(configuration: MLModelConfiguration())
let desc = model.modelDescription

但出现以下错误：

“ MyModel”类型的值没有成员“ modelDescription”

似乎我没有正确使用文档。 感谢您的帮助！

解决方法

MyModel不是MLModel对象，而是专门为您的模型生成的类。但是，它的MLModel属性中确实有一个model对象。

您可以像这样访问描述：

let model = try! MyModel(configuration: MLModelConfiguration())
let desc = model.model.modelDescription
coremlcoremltoolsswiftswiftswiftswift