Skip to content

Mobile Fast Neural Style Transfer in iOS using CoreML with ONNX pre-trained models

License

Notifications You must be signed in to change notification settings

rusito-23/MobileFNST

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Mobile FNST

Mobile Fast Neural Style Transfer

This application was built using the ONNX pretrained models. These ONNX models were converted to iOS 12 compatible CoreML models (using OpSet 8), and then copied in compile time to the application bundle. This process can be found in the scripts.

The application was made using UIKit with MVC Arch, as it's very simple.

These are some screenshots from the app:

Main Screen Candy Pointilism
main candy pointilism
Mosaic Rain Princess Udnie
mosaic rain-princess udnie

These are the styles:

Candy Pointilism Mosaic Rain Princess Udnie
candy pointilism mosaic rain-princess udnie

Run

First, we need to run the script: scripts/models.sh which downloads and converts the models to CoreML. And that's it! Open the .xcodeproj in XCode and run!

Acknowledgements

LICENSE

BSD-3-Clause (as the ONNX Model Zoo)

About

Mobile Fast Neural Style Transfer in iOS using CoreML with ONNX pre-trained models

Resources

License

Stars

Watchers

Forks

Languages