Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,901 Bytes
f499d3b |
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 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
# generate skeleton
config="configs/data/quick_inference.yaml"
require_suffix="obj,fbx,FBX,dae,glb,gltf,vrm"
num_runs=1
force_override="false"
faces_target_count=50000
skeleton_task="configs/task/quick_inference_skeleton_articulationxl_ar_256.yaml"
add_root="false"
seed=12345
npz_dir="tmp"
while [[ "$#" -gt 0 ]]; do
case $1 in
--config) config="$2"; shift ;;
--require_suffix) require_suffix="$2"; shift ;;
--num_runs) num_runs="$2"; shift ;;
--force_override) force_override="$2"; shift ;;
--faces_target_count) faces_target_count="$2"; shift ;;
--skeleton_task) skeleton_task="$2"; shift ;;
--add_root) add_root="$2"; shift ;;
--seed) seed="$2"; shift ;;
--input) input="$2"; shift ;;
--input_dir) input_dir="$2"; shift ;;
--output_dir) output_dir="$2"; shift ;;
--output) output="$2"; shift ;;
*) echo "Unknown parameter: $1"; exit 1 ;;
esac
shift
done
# 1. extract mesh
cmd=" \
bash ./launch/inference/extract.sh \
--config $config \
--require_suffix $require_suffix \
--force_override $force_override \
--num_runs $num_runs \
--faces_target_count $faces_target_count \
"
if [ -n "$input" ]; then
cmd="$cmd --input $input"
fi
if [ -n "$input_dir" ]; then
cmd="$cmd --input_dir $input_dir"
fi
if [ -n "$npz_dir" ]; then
cmd="$cmd --output_dir $npz_dir"
fi
cmd="$cmd &"
eval $cmd
wait
# 2. inference skeleton
cmd="\
python run.py \
--task=$skeleton_task \
--seed=$seed \
"
if [ -n "$input" ]; then
cmd="$cmd --input=$input"
fi
if [ -n "$input_dir" ]; then
cmd="$cmd --input_dir=$input_dir"
fi
if [ -n "$output" ]; then
cmd="$cmd --output=$output"
fi
if [ -n "$output_dir" ]; then
cmd="$cmd --output_dir=$output_dir"
fi
if [ -n "$npz_dir" ]; then
cmd="$cmd --npz_dir=$npz_dir"
fi
eval $cmd
wait
echo "done" |