Swift 在移动端使用CoreML

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

猜你喜欢

转载自www.cnblogs.com/JMarshall/p/11038904.html