Spaces:
Running
Running
File size: 2,002 Bytes
cc48e0c 61bc48a cc48e0c b17fe2a cc48e0c |
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 |
#!/usr/bin/env bash
# bash install.sh --stage 0 --stop_stage 2
verbose=true;
stage=-1
stop_stage=1
# parse options
while true; do
[ -z "${1:-}" ] && break; # break if there are no arguments
case "$1" in
--*) name=$(echo "$1" | sed s/^--// | sed s/-/_/g);
eval '[ -z "${'"$name"'+xxx}" ]' && echo "$0: invalid option $1" 1>&2 && exit 1;
old_value="(eval echo \\$$name)";
if [ "${old_value}" == "true" ] || [ "${old_value}" == "false" ]; then
was_bool=true;
else
was_bool=false;
fi
# Set the variable to the right value-- the escaped quotes make it work if
# the option had spaces, like --cmd "queue.pl -sync y"
eval "${name}=\"$2\"";
# Check that Boolean-valued arguments are really Boolean.
if $was_bool && [[ "$2" != "true" && "$2" != "false" ]]; then
echo "$0: expected \"true\" or \"false\": $1 $2" 1>&2
exit 1;
fi
shift 2;
;;
*) break;
esac
done
work_dir="$(pwd)"
thirdparty_dir="$(pwd)/thirdparty"
pretrained_models_dir="$(pwd)/pretrained_models"
silero_vad_model_dir="${pretrained_models_dir}/silero_vad"
mkdir -p "${thirdparty_dir}"
mkdir -p "${pretrained_models_dir}"
mkdir -p "${silero_vad_model_dir}"
if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
$verbose && echo "stage 0: install onnxruntime"
cd "${thirdparty_dir}"
# install onnxruntime
# https://onnxruntime.ai/docs/install/
# https://github.com/microsoft/onnxruntime/releases
wget https://github.com/microsoft/onnxruntime/releases/download/v1.20.1/onnxruntime-linux-x64-1.20.1.tgz
tar -zxvf onnxruntime-linux-x64-1.20.1.tgz
fi
if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
$verbose && echo "stage 1: download silero_vad.onnx"
cd "${silero_vad_model_dir}"
# download model
# https://github.com/snakers4/silero-vad/tree/master/src/silero_vad/data
wget https://github.com/snakers4/silero-vad/raw/refs/heads/master/src/silero_vad/data/silero_vad.onnx
fi
|