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

Question

我有一个 CoreML model，我已经使用 coremltools 向 model 添加了信息：

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

有没有办法从 Swift 访问此信息？

在 Apple 的文档中，它是这样写的：

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

我试过这个：

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

但我收到以下错误：

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

看来我没有正确使用文档。 谢谢你的帮助！

Answer 1

MyModel不是MLModel object，而是专门为您的 model 生成的 class。但是，它在model属性中确实有一个MLModel object。

您可以像这样访问描述：

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