/style-me

Client-server iOS app for generation of artistic representations :art:

Primary LanguageSwiftMIT LicenseMIT

Style Me!

The project aims to implement an iOS application for Neural Style Transfer on photos of people.

Functionality

  • User can either take a photo with the front/back camera or upload one from the photo gallery.
  • One can generate a stylized version of the uploaded photo.
  • Generated photo can be saved on the device.

How does it work?

  • Binary classification model (human/not human) implemented and trained with Keras, and converted into CoreML format, which allows performing classification of client-side.

stylization

  • The PyTorch model performs Neural Style Transfer as a part of the web-service, implemented with aiohttp.

human-not-human

Versioning

SemVer is used for versioning. For the versions available, see the tags on this repository.

License

This project is licensed under the MIT License - see the LICENSE file for details.

Authors

  • Sergey Kastryulin - Initial work - snk4tr