25 lines
485 B
Bash
25 lines
485 B
Bash
#!/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
|