Skip to content
This repository has been archived by the owner on Feb 15, 2020. It is now read-only.

backnotprop/fast-style-transfer-coreml

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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)