Neural Style Transfer on the browser

This repo is a JavaScript version of Tensorflow's Neural Style Transfer tutorial, developed based on the original style-transfer algorithm outlined in A Neural Algorithm of Artistic Style (Gatys et al.)

Demo Instance

Click here.

Deployment

Running the app on an EC2 instance

Setting up an EC2 instance

Create an EC2 instance with Amazon Linux 2 AMI
SSH into the instance
Install docker and docker-compose

## Install docker
sudo yum install docker

## Start docker daemon
sudo systemctl start docker

## Install docker-compose
sudo curl -L https://github.com/docker/compose/releases/download/1.26.2/docker-compose-`uname -s`-`uname -m` -o /usr/local/bin/docker-compose
sudo chmod +x /usr/local/bin/docker-compose

## Create the docker group 
sudo groupadd docker

## Add your user to the docker group
sudo usermod -aG docker ${USER}

## Verify that you can run docker commands without sudo
## (you may need to exit first then SSH again)
docker ps

Install git: sudo yum install git
Clone the project repo

Running the app

cd <project directory>/NeuralStyleTransfer
Change the base URL in docker-compose.yml to the VM's public IP
Run docker-compose up --build
Navigate to <vm public IP>:3000 on your browser

Limitations

Your browser is going to be really slow when the app is running 😅.