Spaces:
Sleeping
Sleeping
File size: 1,979 Bytes
054b62d 351f98b 054b62d b5e054c 054b62d b5e054c 054b62d 9e5c7f1 054b62d |
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 |
#!/usr/bin/env bash
# bash install.sh --stage -1 --stop_stage -1 --system_version windows
# bash install.sh --stage 0 --stop_stage 0 --system_version ubuntu
python_version=3.8.10
system_version=ubuntu
verbose=true;
stage=-1
stop_stage=0
# 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)"
data_dir="$(pwd)/data"
trained_models_dir="$(pwd)/trained_models"
mkdir -p "${data_dir}"
mkdir -p "${trained_models_dir}"
if [ ${system_version} = "centos" ]; then
yum install -y bzip2 git lrzsz wget vim cronie
elif [ ${system_version} = "ubuntu" ]; then
apt-get update
apt-get install -y bzip2 git lrzsz wget vim cron
fi
if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
$verbose && echo "stage 0: download deep filter net models."
df_models_dir="${trained_models_dir}/df"
mkdir -p "${df_models_dir}"
cd "${df_models_dir}" || exit 1;
model_name_array=(
DeepFilterNet
DeepFilterNet2
DeepFilterNet3
)
for model_name in ${model_name_array[*]}
do
if [ ! -e "${model_name}.zip" ]; then
wget -c "https://github.com/Rikorose/DeepFilterNet/raw/main/models/${model_name}.zip"
fi
done
fi
|