device='/gpu:0'

# Parse arguments
content_image="$1"
content_dir=$(dirname "$content_image")
content_filename=$(basename "$content_image")

style_image="$2"
style_dir=$(dirname "$style_image" )
style_filename=$(basename "$style_image")

echo "Rendering stylized image. This may take a while..."
python neural_style.py \
       --content_img "${content_filename}" \
       --content_img_dir "${content_dir}" \
       --style_imgs "${style_filename}" \
       --style_imgs_dir "${style_dir}" \
       --device "${device}" \
       --verbose;

#mkdir $3_color
#cp image_output/result/* $3_color/
#
#python neural_style.py \
#       --content_img "${content_filename}" \
#       --content_img_dir "${content_dir}" \
#       --style_imgs "${style_filename}" \
#       --style_imgs_dir "${style_dir}" \
#       --device "${device}" \
#       --original_colors \
#       --verbose;
#
mkdir $3
cp image_output/result/* $3/
cp image_output/result/result.png $3.png