#!/bin/bash

INPUT="${1}"
STYLE="${2}"
OUTPUT="${3}"

# https://github.com/ProGamerGov/neural-style-pt

neural-style \
	-style_image "${STYLE}" \
	-content_image "${INPUT}" \
	-output_image "${OUTPUT}" \
	-model_file python/nin_imagenet.pth \
	-gpu c \
	-backend nn \
	-num_iterations 500 \
	-seed 123 \
	-content_layers relu0,relu3,relu7,relu12 \
	-style_layers relu0,relu3,relu7,relu12 \
	-content_weight 10 \
	-style_weight 500 \
	-save_iter 0 \
	-image_size 720 \
	-optimizer adam