码迷,mamicode.com
首页 > 移动开发 > 详细

Swift 在移动端使用CoreML

时间:2019-06-17 12:56:10      阅读:127      评论:0      收藏:0      [点我收藏+]

标签:图像   特定   新特性   文件   ima   支持   输入输出   mac   swift   

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

Swift 在移动端使用CoreML

标签:图像   特定   新特性   文件   ima   支持   输入输出   mac   swift   

原文地址:https://www.cnblogs.com/JMarshall/p/11038904.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!