blob: e325367223b394e04ef703043968189275db08e9 (
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
|
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
|
