美文网首页
深度学习应用:iOS 上的图像风格迁移

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

作者: iOSDevLog | 来源:发表于2019-01-06 23:38 被阅读65次
    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

    相关文章

      网友评论

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

          本文链接:https://www.haomeiwen.com/subject/clwerqtx.html