深度学习应用:iOS 上的图像风格迁移

2019-01-06  本文已影响65人  iOSDevLog
fast-style-transfer-coreml

图像风格迁移,用 python 就可以实现,如果想要在手机上面(不联网)查看效果怎么办呢?

如果你是用 iOS 系统,你一定听说过 Prisma,它赢得了 2016 年度最佳应用程序,就是这样,它在短短几秒钟内,可以将你的图片转换成你所选择的任何风格。

Prisma.png

这里我们使用 iOS 11 推出的 CoreML 实现 Prisma 类似的功能。

Android版的见 tensorflow 官方提供的例子:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/android/

TF Stylize

首先需要用 Tensorflow 训练好模型,之后可以用 Apple 官方提供的转换工具 coremltools 导出成 iOS 11 支持的 CoreML 格式。

新建 StyleArts 项目,将导出的 *.mlmodel 文件拖入项目。

具体实现细节可以参考我改的代码 https://github.com/iOSDevLog/StyleArts 或者 GitHub 上面其它的实现。

StyleArts.PNG

移动端虽然不适合训练机器学习模型,不过可以围魏救赵,通过导出 PC 上面训练好的模型也可以体验人工智能带来的便利。

想体验 StyleArts AppStore 效果的可在 App Store 里面搜索 StyleArts

StyleArtsAppStore.PNG
上一篇下一篇

猜你喜欢

热点阅读