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的主要内容,如果未能解决你的问题,请参考以下文章