backnotprop / fast-style-transfer-coreml

Home Page:https://medium.com/@rambossa/diy-prisma-fast-style-transfer-app-with-coreml-and-tensorflow-817c3b90dacd

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Fast Style Transfer, CoreML, TensorFlow

Create an iOS app — like Prisma — with CoreML, Fast Style Transfer, and TensorFlow.

Installation

git clone https://github.com/mdramos/fast-style-transfer-coreml.git

Dependencies

Note that you'll want to make the necessary adjustments for Fast-Style-Transfer and TensorFlow-CoreML. I write about these in my medium article. The adjusments are also available in this repo: fst and tf-coreml.

Models

You can train your own models using FST, and use my adjustments to convert them to CoreML models. I've already converted the pre-trained models from FST:

CoreML Models: https://drive.google.com/drive/folders/1CBSanBHbXC5-bJNTTk3-r1WSq56z0eKG?usp=sharing

iOS

Once you have your models, just import them into the Xcode project. The current setup looks for a particular set of models (wave, udnie, rain_princess, and la_muse)

License

Copyright (c) 2018 Michael Ramos. MIT on any of my code, but you'll need to abide by the Licences of the libraries used (FST, tf-coreml)

About

https://medium.com/@rambossa/diy-prisma-fast-style-transfer-app-with-coreml-and-tensorflow-817c3b90dacd

License:MIT License


Languages

Language:Python 94.5%Language:Swift 5.3%Language:Shell 0.1%