romanmazeev / imagestyler

iOS/MacOS app that applying one of the proposed styles to selected image

Image Styler

Image Styler changes the image selected by the user by applying one of the proposed styles to it using a neural network.

Screenshots

iOS	MacOS

Requirements

• Xcode 11
• Swift 5
• iOS 13
• MacOS 10.15

For model training

• Python 3.6
• TuriCreate

How to train own model

For training, the TuriCreate framework is used. To train your own model you need:

1. Move your styles images to style folder.
2. Move the training dataset into content folder. Photos should be similar to those that you will use in the application to apply style. I used for training a selfie dataset.
3. [Optional] Move the test dataset into test folder.
4. Run this script script to start training.

Remember to change the styles data in the application.