Skip to content

snk4tr/style-me

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

39 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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