blob: fe9f707284d1c7e67478dbb1d08a5b016de63274 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
set -e
# Get a carriage return into `cr`
cr=`echo $'\n.'`
cr=${cr%.}
#if [ "$#" -le 1 ]; then
# echo "Usage: bash stylize_image.sh <path_to_content_image> <path_to_style_image>"
# exit 1
#fi
#
#echo ""
#read -p "Did you install the required dependencies? [y/n] $cr > " dependencies
#
#if [ "$dependencies" != "y" ]; then
# echo "Error: Requires dependencies: tensorflow, opencv2 (python), scipy"
# exit 1;
#fi
#
#echo ""
#read -p "Do you have a CUDA enabled GPU? [y/n] $cr > " cuda
#
#if [ "$cuda" != "y" ]; then
# device='/cpu:0'
#else
device='/gpu:0'
#fi
# 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;
|
