nixpkgs/pkgs/tools/graphics/fast-neural-doodle/fast-neural-doodle.sh

29 lines
788 B
Bash
Raw Normal View History

#! @bash@/bin/bash
declare -a args
c=1
flag=
for arg in "$@"; do
if test "$arg" = "${arg#-}" && test "$arg" = "${arg#/}" && test -n "$flag"; then
arg="$PWD/$arg"
flag=
elif (test "$arg" != "${arg%_image}" || test "$arg" == "-masks_hdf5") && test "$arg" != "${arg#-}"; then
flag=1
else
flag=
fi
args[c]="$arg";
c=$((c+1));
done
cd "@out@/lib/lua/fast_neural_doodle"
export LUA_PATH="$LUA_PATH${LUA_PATH:+;}@loadcaffe@/lua/?/init.lua;@loadcaffe@/lua/?.lua"
export LUA_PATH="$LUA_PATH${LUA_PATH:+;}@torch_hdf5@/lua/?/init.lua;@torch_hdf5@/lua/?.lua"
export LUA_CPATH="$LUA_CPATH${LUA_CPATH:+;}@loadcaffe@/lib/?.so"
set -x
@torch@/bin/th fast_neural_doodle.lua "${args[@]}"