微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

如何解决如何在CoreML中访问模型的描述?

我有一个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

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。