| #!/bin/bash |
| |
| MXNET_ROOT=$(cd "$(dirname $0)/../../.."; pwd) |
| CLASS_PATH=$MXNET_ROOT/scala-package/assembly/linux-x86_64-gpu/target/*:$MXNET_ROOT/scala-package/examples/target/*:$MXNET_ROOT/scala-package/examples/target/classes/lib/* |
| INPUT_IMG=$1 |
| STYLE_IMG=$2 |
| MODEL_PATH=$MXNET_ROOT/example/neural-style/model/vgg19.params |
| OUTPUT_DIR=$MXNET_ROOT/example/neural-style/output |
| |
| java -Xmx1024m -cp $CLASS_PATH \ |
| ml.dmlc.mxnetexamples.neuralstyle.NeuralStyle \ |
| --content-image $INPUT_IMG \ |
| --style-image $STYLE_IMG \ |
| --model-path $MODEL_PATH \ |
| --output-dir $OUTPUT_DIR |
