//AI Photobooth

Project created:
Post updated:
Cover Image for AI Photobooth
pythonaicomputer visionphotography

My dear friend, if I can I'll find you again.

I was contracted by Edelman to design and create an interactive exhibit for their AI month - one that would utilize neural style transfer to provide a fun experience for visitors to their NYC headquarters. I went with the direction of film and pop culture, and thought it would be interesting for people to see themselves transformed into styles from popular media references. I trained 5 different models, 3 of which were chosen for show: Spirited Away, Sin City, Game of Thrones, Blade Runner, and Alice in Wonderland.

Video demo of AI Photobooth UI Video demo of UI.

A Python server with a REST API powers the backend and performs the image transformations. After trying the experience, users can enter in their emails and have the photos sent to them via Sendgrid's API. Images can be taken with a web camera or with a mobile / tablet camera (when accessing the website on a mobile device).

I used code samples and research from ywng's multi-style-transfer, which was a modified implementation of fast-neural-style to allow for multiple images to be used as a reference for style transfer. Although the original purpose of their design was to allow for different types of styles to be transferred, I found that this algorithm also greatly improved results for just a single style transfer (with multiple images of the same style used as reference).

Essentially, I would capture various frames from the sources of media I had, all somewhat cohesive in aesthetic but different in the image content, and trained models those groups of images for a style instead of just a singular one. This made the results of image transformations a lot more consistent across a wider spectrum input images, as it was able to create a "average" sense of details to transform from the data. I had also incorporated Labeled Faces in the Wild dataset within the base training data, to give greater accuracy for portraits and selfie transformations.