Swift 在移动端使用CoreML

Posted jmarshall

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Swift 在移动端使用CoreML相关的知识,希望对你有一定的参考价值。

  从PC端搞到你要用的训练好的神经网络,首先就死固化你的模型得到pb文件,如果实在tensorflow的框架下跑的,使用tfcoreml工具进行转化就ok,目前CoreML3的新特性还在开发支持中,有一些错误有望在新的版本中修复。得到.coreml文件的关键在于知道输入输出层的参数名字,然后按照得到得值进行预处理。我因为用到的都是Image作为输入的,所以需要设置特定的参数进行处理。可以参考:https://machinethink.net/blog/help-core-ml-gives-wrong-output/ 里面有设置不同的图像预处理,以及得到的数据使用规范。坑真的多,但是模型能够运行的时候真的amazing 是我独享的moment

以上是关于Swift 在移动端使用CoreML的主要内容,如果未能解决你的问题,请参考以下文章

在 Swift 中设置 CoreML 模型 - 图像分类器

如何将 CoreML 模型添加到 Swift 包中?

swift Camera_PIcker_CoreML

Swift 中 Vision/CoreML 对象识别器的精度

Swift中Vision / CoreML对象识别器的精度

CoreML 推理时间是不是会随着迭代而增长?