如何在 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