
fast-style-transfer-coreml
圖像風格遷移,用 python 就可以實現(xiàn),如果想要在手機上面(不聯(lián)網(wǎng))查看效果怎么辦呢?
如果你是用 iOS 系統(tǒng),你一定聽說過 Prisma,它贏得了 2016 年度最佳應用程序,就是這樣,它在短短幾秒鐘內(nèi),可以將你的圖片轉(zhuǎn)換成你所選擇的任何風格。

Prisma.png
這里我們使用 iOS 11 推出的 CoreML 實現(xiàn) Prisma 類似的功能。
Android版的見 tensorflow 官方提供的例子:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/android/

TF Stylize
首先需要用 Tensorflow 訓練好模型,之后可以用 Apple 官方提供的轉(zhuǎn)換工具 coremltools 導出成 iOS 11 支持的 CoreML 格式。
新建 StyleArts 項目,將導出的 *.mlmodel 文件拖入項目。
具體實現(xiàn)細節(jié)可以參考我改的代碼 https://github.com/iOSDevLog/StyleArts 或者 GitHub 上面其它的實現(xiàn)。
StyleArts.PNG
移動端雖然不適合訓練機器學習模型,不過可以圍魏救趙,通過導出 PC 上面訓練好的模型也可以體驗人工智能帶來的便利。
想體驗 StyleArts AppStore 效果的可在 App Store 里面搜索 StyleArts。
StyleArtsAppStore.PNG