Skip to content

pauliver/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)

Releases

No releases published

Packages

No packages published

Languages

  • Python 94.6%
  • Swift 5.3%
  • Shell 0.1%