From 580b11b57ac4b62f7e2acda73813a4e10e8e4cd3 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 10 十月 2023 17:17:29 +0800
Subject: [PATCH] v0.8.0
---
funasr/runtime/deploy_tools/funasr-runtime-deploy-offline-cpu-zh.sh | 1082 +++++++++++++++++++++++++++++++++++++++-----------------
1 files changed, 745 insertions(+), 337 deletions(-)
diff --git a/funasr/runtime/deploy_tools/funasr-runtime-deploy-offline-cpu-zh.sh b/funasr/runtime/deploy_tools/funasr-runtime-deploy-offline-cpu-zh.sh
index 813373d..5738d6e 100644
--- a/funasr/runtime/deploy_tools/funasr-runtime-deploy-offline-cpu-zh.sh
+++ b/funasr/runtime/deploy_tools/funasr-runtime-deploy-offline-cpu-zh.sh
@@ -1,7 +1,7 @@
#!/usr/bin/env bash
-scriptVersion="0.0.4"
-scriptDate="20230701"
+scriptVersion="0.0.7"
+scriptDate="20230818"
# Set color
@@ -22,16 +22,22 @@
UNDERLINE="\033[4m"
# Current folder
-cur_dir=`pwd`
+CUR_DIR=`pwd`
+SUDO_CMD="sudo"
-DEFAULT_DOCKER_OFFLINE_CPU_ZH_LISTS="https://raw.githubusercontent.com/alibaba-damo-academy/FunASR/main/funasr/runtime/docs/docker_offline_cpu_zh_lists"
-DEFAULT_DOCKER_IMAGE_LISTS=$DEFAULT_DOCKER_OFFLINE_CPU_ZH_LISTS
+DEFAULT_DOCKER_OFFLINE_CPU_ZH_LISTS_OSS="https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/docker_lists/docker_offline_cpu_zh_lists"
+DEFAULT_DOCKER_OFFLINE_CPU_ZH_LISTS_GIT="https://raw.githubusercontent.com/alibaba-damo-academy/FunASR/main/funasr/runtime/docs/docker_offline_cpu_zh_lists"
+DEFAULT_DOCKER_IMAGE_LISTS=$DEFAULT_DOCKER_OFFLINE_CPU_ZH_LISTS_OSS
DEFAULT_FUNASR_DOCKER_URL="registry.cn-hangzhou.aliyuncs.com/funasr_repo/funasr"
-DEFAULT_FUNASR_RUNTIME_SDK_NAME="funasr-runtime-sdk"
-DEFAULT_FUNASR_LOCAL_WORKSPACE="${cur_dir}/${DEFAULT_FUNASR_RUNTIME_SDK_NAME}"
-DEFAULT_FUNASR_CONFIG_DIR="/var/funasr"
+DEFAULT_FUNASR_RUNTIME_RESOURCES="funasr-runtime-resources"
+DEFAULT_FUNASR_LOCAL_WORKSPACE=${CUR_DIR}/${DEFAULT_FUNASR_RUNTIME_RESOURCES}
+DEFAULT_FUNASR_CONFIG_DIR=""
+DEFAULT_FUNASR_CONFIG_DIR_BAK="/var/funasr"
DEFAULT_FUNASR_CONFIG_FILE="${DEFAULT_FUNASR_CONFIG_DIR}/config"
+DEFAULT_FUNASR_SERVER_CONFIG_FILE="${DEFAULT_FUNASR_CONFIG_DIR}/server_config"
+DEFAULT_FUNASR_PROGRESS_TXT="${DEFAULT_FUNASR_CONFIG_DIR}/progress.txt"
+DEFAULT_FUNASR_SERVER_LOG="${DEFAULT_FUNASR_CONFIG_DIR}/server_console.log"
DEFAULT_FUNASR_WORKSPACE_DIR="/workspace/models"
DEFAULT_DOCKER_PORT="10095"
DEFAULT_PROGRESS_FILENAME="progress.txt"
@@ -39,32 +45,35 @@
DEFAULT_DOCKER_EXEC_DIR="/workspace/FunASR/funasr/runtime/websocket/build/bin"
DEFAULT_DOCKER_EXEC_PATH=${DEFAULT_DOCKER_EXEC_DIR}/${DEFAULT_SERVER_EXEC_NAME}
DEFAULT_SAMPLES_NAME="funasr_samples"
+DEFAULT_SAMPLES_DIR="samples"
DEFAULT_SAMPLES_URL="https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/sample/${DEFAULT_SAMPLES_NAME}.tar.gz"
-
SAMPLE_CLIENTS=( \
"Python" \
"Linux_Cpp" \
)
DOCKER_IMAGES=()
-PROGRESS_TXT="/var/funasr/progress.txt"
-ASR_PERCENT_INT=0
-VAD_PERCENT_INT=0
-PUNC_PERCENT_INT=0
-ASR_TITLE="Downloading"
-ASR_PERCENT="0"
-ASR_SPEED="0KB/s"
-ASR_REVISION=""
-VAD_TITLE="Downloading"
-VAD_PERCENT="0"
-VAD_SPEED="0KB/s"
-VAD_REVISION=""
-PUNC_TITLE="Downloading"
-PUNC_PERCENT="0"
-PUNC_SPEED="0KB/s"
-PUNC_REVISION=""
+ASR_MODELS=()
+VAD_MODELS=()
+PUNC_MODELS=()
-ServerProgress(){
+# Handles the download progress bar
+asr_percent_int=0
+vad_percent_int=0
+punc_percent_int=0
+asr_title="Downloading"
+asr_percent="0"
+asr_speed="0KB/s"
+asr_revision=""
+vad_title="Downloading"
+vad_percent="0"
+vad_speed="0KB/s"
+vad_revision=""
+punc_title="Downloading"
+punc_percent="0"
+punc_speed="0KB/s"
+punc_revision=""
+serverProgress(){
status_flag="STATUS:"
stage=0
wait=0
@@ -72,7 +81,7 @@
while true
do
- if [ -f "$PROGRESS_TXT" ]; then
+ if [ -f "$DEFAULT_FUNASR_PROGRESS_TXT" ]; then
break
else
sleep 1
@@ -83,8 +92,8 @@
fi
done
- if [ ! -f "$PROGRESS_TXT" ]; then
- echo -e " ${RED}The note of progress does not exist.($PROGRESS_TXT) ${PLAIN}"
+ if [ ! -f "$DEFAULT_FUNASR_PROGRESS_TXT" ]; then
+ echo -e " ${RED}The note of progress does not exist.($DEFAULT_FUNASR_PROGRESS_TXT) ${PLAIN}"
return 98
fi
@@ -116,22 +125,22 @@
fi
result=$(echo $line | grep "title:")
if [ "$result" != "" ]; then
- ASR_TITLE=${line#*:}
+ asr_title=${line#*:}
continue
fi
result=$(echo $line | grep "percent:")
if [ "$result" != "" ]; then
- ASR_PERCENT=${line#*:}
+ asr_percent=${line#*:}
continue
fi
result=$(echo $line | grep "speed:")
if [ "$result" != "" ]; then
- ASR_SPEED=${line#*:}
+ asr_speed=${line#*:}
continue
fi
result=$(echo $line | grep "revision:")
if [ "$result" != "" ]; then
- ASR_REVISION=${line#*:}
+ asr_revision=${line#*:}
continue
fi
elif [ $stage -eq 4 ]; then
@@ -142,22 +151,22 @@
fi
result=$(echo $line | grep "title:")
if [ "$result" != "" ]; then
- VAD_TITLE=${line#*:}
+ vad_title=${line#*:}
continue
fi
result=$(echo $line | grep "percent:")
if [ "$result" != "" ]; then
- VAD_PERCENT=${line#*:}
+ vad_percent=${line#*:}
continue
fi
result=$(echo $line | grep "speed:")
if [ "$result" != "" ]; then
- VAD_SPEED=${line#*:}
+ vad_speed=${line#*:}
continue
fi
result=$(echo $line | grep "revision:")
if [ "$result" != "" ]; then
- VAD_REVISION=${line#*:}
+ vad_revision=${line#*:}
continue
fi
elif [ $stage -eq 5 ]; then
@@ -169,42 +178,42 @@
fi
result=$(echo $line | grep "title:")
if [ "$result" != "" ]; then
- PUNC_TITLE=${line#*:}
+ punc_title=${line#*:}
continue
fi
result=$(echo $line | grep "percent:")
if [ "$result" != "" ]; then
- PUNC_PERCENT=${line#*:}
+ punc_percent=${line#*:}
continue
fi
result=$(echo $line | grep "speed:")
if [ "$result" != "" ]; then
- PUNC_SPEED=${line#*:}
+ punc_speed=${line#*:}
continue
fi
result=$(echo $line | grep "revision:")
if [ "$result" != "" ]; then
- PUNC_REVISION=${line#*:}
+ punc_revision=${line#*:}
continue
fi
elif [ $stage -eq 99 ]; then
echo -e " ${RED}ERROR: $line${PLAIN}"
fi
- done < $PROGRESS_TXT
+ done < $DEFAULT_FUNASR_PROGRESS_TXT
if [ $stage -ne 99 ]; then
- DrawProgress "ASR " $ASR_TITLE $ASR_PERCENT $ASR_SPEED $ASR_REVISION $ASR_PERCENT_INT
- ASR_PERCENT_INT=$?
- DrawProgress "VAD " $VAD_TITLE $VAD_PERCENT $VAD_SPEED $VAD_REVISION $VAD_PERCENT_INT
- VAD_PERCENT_INT=$?
- DrawProgress "PUNC" $PUNC_TITLE $PUNC_PERCENT $PUNC_SPEED $PUNC_REVISION $PUNC_PERCENT_INT
- PUNC_PERCENT_INT=$?
+ drawProgress "ASR " $asr_title $asr_percent $asr_speed $asr_revision $asr_percent_int
+ asr_percent_int=$?
+ drawProgress "VAD " $vad_title $vad_percent $vad_speed $vad_revision $vad_percent_int
+ vad_percent_int=$?
+ drawProgress "PUNC" $punc_title $punc_percent $punc_speed $punc_revision $punc_percent_int
+ punc_percent_int=$?
fi
return $stage
}
-DrawProgress(){
+drawProgress(){
model=$1
title=$2
percent_str=$3
@@ -221,8 +230,8 @@
fi
fi
- LOADING_FLAG="Loading"
- if [ "$title" = "$LOADING_FLAG" ]; then
+ loading_flag="Loading"
+ if [ "$title" = "$loading_flag" ]; then
progress=100
fi
@@ -237,9 +246,9 @@
let color=36
let index=max*2
if [ -z "$speed" ]; then
- printf "\r \e[0;$color;1m[%s][%-11s][%-50s][%d%%][%s]\e[0m" "$model" "$title" "$str" "$$index" "$revision"
+ printf "\r \e[0;${CYAN}[%s][%-11s][%-50s][%d%%][%s]\e[0m" "$model" "$title" "$str" "$$index" "$revision"
else
- printf "\r \e[0;$color;1m[%s][%-11s][%-50s][%3d%%][%8s][%s]\e[0m" "$model" "$title" "$str" "$index" "$speed" "$revision"
+ printf "\r \e[0;${CYAN}[%s][%-11s][%-50s][%3d%%][%8s][%s]\e[0m" "$model" "$title" "$str" "$index" "$speed" "$revision"
fi
printf "\n"
@@ -288,20 +297,71 @@
return $result
}
+
+full_path=""
+relativePathToFullPath(){
+ relativePath=$1
+ firstChar=${relativePath: 0: 1}
+ if [[ "$firstChar" == "" ]]; then
+ full_path=$relativePath
+ elif [[ "$firstChar" == "/" ]]; then
+ full_path=$relativePath
+ fi
+
+ tmpPath1=`dirname $relativePath`
+ tmpFullpath1=`cd $tmpPath1 && pwd`
+ tmpPath2=`basename $relativePath`
+ full_path=${tmpFullpath1}/${tmpPath2}
+}
+
initConfiguration(){
- mkdir -p $DEFAULT_FUNASR_CONFIG_DIR
+ if [ -z "$DEFAULT_FUNASR_CONFIG_DIR" ];then
+ DEFAULT_FUNASR_CONFIG_DIR="$HOME"
+ if [ -z "$DEFAULT_FUNASR_CONFIG_DIR" ];then
+ $DEFAULT_FUNASR_CONFIG_DIR=$(echo ~/)
+ if [ -z "$DEFAULT_FUNASR_CONFIG_DIR" ];then
+ $DEFAULT_FUNASR_CONFIG_DIR=$DEFAULT_FUNASR_CONFIG_DIR_BAK
+ fi
+ fi
+ DEFAULT_FUNASR_CONFIG_DIR=${DEFAULT_FUNASR_CONFIG_DIR}/.funasr
+ fi
+
+ if [ ! -z "$DEFAULT_FUNASR_CONFIG_DIR" ]; then
+ $SUDO_CMD mkdir -p $DEFAULT_FUNASR_CONFIG_DIR
+ else
+ echo -e " ${RED}DEFAULT_FUNASR_CONFIG_DIR is empty!${PLAIN}"
+ exit 1
+ fi
+ if [ ! -d "$DEFAULT_FUNASR_CONFIG_DIR" ]; then
+ echo -e " ${RED}${DEFAULT_FUNASR_CONFIG_DIR} does not exist!${PLAIN}"
+ exit 2
+ fi
+
+ DEFAULT_FUNASR_CONFIG_FILE="${DEFAULT_FUNASR_CONFIG_DIR}/config"
+ DEFAULT_FUNASR_SERVER_CONFIG_FILE="${DEFAULT_FUNASR_CONFIG_DIR}/server_config"
+ DEFAULT_FUNASR_PROGRESS_TXT="${DEFAULT_FUNASR_CONFIG_DIR}/progress.txt"
+ DEFAULT_FUNASR_SERVER_LOG="${DEFAULT_FUNASR_CONFIG_DIR}/server_console.log"
+
if [ ! -f $DEFAULT_FUNASR_CONFIG_FILE ]; then
- touch $DEFAULT_FUNASR_CONFIG_FILE
+ $SUDO_CMD touch $DEFAULT_FUNASR_CONFIG_FILE
+ fi
+ if [ ! -f $DEFAULT_FUNASR_SERVER_CONFIG_FILE ]; then
+ $SUDO_CMD touch $DEFAULT_FUNASR_SERVER_CONFIG_FILE
fi
}
initParameters(){
# Init workspace in local by new parameters.
- PARAMS_FUNASR_SAMPLES_LOCAL_DIR=${PARAMS_FUNASR_LOCAL_WORKSPACE}/${DEFAULT_SAMPLES_NAME}
+ PARAMS_FUNASR_SAMPLES_LOCAL_PATH=${PARAMS_FUNASR_LOCAL_WORKSPACE}/${DEFAULT_SAMPLES_NAME}.tar.gz
+ PARAMS_FUNASR_SAMPLES_LOCAL_DIR=${PARAMS_FUNASR_LOCAL_WORKSPACE}/${DEFAULT_SAMPLES_DIR}
PARAMS_FUNASR_LOCAL_MODELS_DIR="${PARAMS_FUNASR_LOCAL_WORKSPACE}/models"
- mkdir -p $PARAMS_FUNASR_LOCAL_WORKSPACE
- mkdir -p $PARAMS_FUNASR_LOCAL_MODELS_DIR
+ if [ ! -z "$PARAMS_FUNASR_LOCAL_WORKSPACE" ]; then
+ mkdir -p $PARAMS_FUNASR_LOCAL_WORKSPACE
+ fi
+ if [ ! -z "$PARAMS_FUNASR_LOCAL_MODELS_DIR" ]; then
+ mkdir -p $PARAMS_FUNASR_LOCAL_MODELS_DIR
+ fi
}
# Parse the parameters from the docker list file.
@@ -320,41 +380,45 @@
# Get a list of docker images.
readDockerInfoFromUrl(){
- list_url=$DEFAULT_DOCKER_IMAGE_LISTS
while true
do
+ list_url=$DEFAULT_DOCKER_IMAGE_LISTS
content=$(curl --connect-timeout 10 -m 10 -s $list_url)
if [ ! -z "$content" ]; then
break
else
echo -e " ${RED}Unable to get docker image list due to network issues, try again.${PLAIN}"
+
+ # switch sources of docker image lists
+ if [ "$list_url" = "$DEFAULT_DOCKER_OFFLINE_CPU_ZH_LISTS_OSS" ]; then
+ DEFAULT_DOCKER_IMAGE_LISTS=$DEFAULT_DOCKER_OFFLINE_CPU_ZH_LISTS_GIT
+ else
+ DEFAULT_DOCKER_IMAGE_LISTS=$DEFAULT_DOCKER_OFFLINE_CPU_ZH_LISTS_OSS
+ fi
fi
done
array=($(echo "$content"))
len=${#array[@]}
- stage=0
- docker_flag="DOCKER:"
- judge_flag=":"
for i in ${array[@]}
do
findTypeOfDockerInfo $i
if [ "$docker_info_cur_key" = "DOCKER:" ]; then
if [ ! -z "$docker_info_cur_val" ]; then
docker_name=${DEFAULT_FUNASR_DOCKER_URL}:${docker_info_cur_val}
- DOCKER_IMAGES[${#DOCKER_IMAGES[*]}]=$docker_name
+ DOCKER_IMAGES[${#DOCKER_IMAGES[*]}]=$docker_name
fi
elif [ "$docker_info_cur_key" = "DEFAULT_ASR_MODEL:" ]; then
if [ ! -z "$docker_info_cur_val" ]; then
- PARAMS_ASR_ID=$docker_info_cur_val
+ ASR_MODELS[${#ASR_MODELS[*]}]=$docker_info_cur_val
fi
elif [ "$docker_info_cur_key" = "DEFAULT_VAD_MODEL:" ]; then
if [ ! -z "$docker_info_cur_val" ]; then
- PARAMS_VAD_ID=$docker_info_cur_val
+ VAD_MODELS[${#VAD_MODELS[*]}]=$docker_info_cur_val
fi
elif [ "$docker_info_cur_key" = "DEFAULT_PUNC_MODEL:" ]; then
if [ ! -z "$docker_info_cur_val" ]; then
- PARAMS_PUNC_ID=$docker_info_cur_val
+ PUNC_MODELS[${#PUNC_MODELS[*]}]=$docker_info_cur_val
fi
fi
done
@@ -365,73 +429,108 @@
rootNess(){
echo -e "${UNDERLINE}${BOLD}[0/6]${PLAIN}"
echo -e " ${YELLOW}Please check root access.${PLAIN}"
- echo
echo -e " ${WARNING} MUST RUN AS ${RED}ROOT${PLAIN} USER!"
if [[ $EUID -ne 0 ]]; then
echo -e " ${ERROR} MUST RUN AS ${RED}ROOT${PLAIN} USER!"
fi
- cd $cur_dir
- echo
-}
-
-# Set the host working path.
-setupLocalWorkspaceDir(){
- echo -e "${UNDERLINE}${BOLD}[1/6]${PLAIN}"
- params_local_workspace=`sed '/^PARAMS_FUNASR_LOCAL_WORKSPACE=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- if [ -z "$params_local_workspace" ]; then
- params_local_workspace=$DEFAULT_FUNASR_LOCAL_WORKSPACE
+ check_sudo=$(which sudo | wc -l)
+ if [ $check_sudo -eq 0 ]; then
+ SUDO_CMD=""
fi
- while true
- do
- echo -e " ${YELLOW}Please enter the local path of workspace.${PLAIN}"
- echo -e " Setting the local path of workspace, default(${CYAN}${params_local_workspace}${PLAIN}): \c"
- read PARAMS_FUNASR_LOCAL_WORKSPACE
- if [ -z "$PARAMS_FUNASR_LOCAL_WORKSPACE" ]; then
- if [ -z "$params_local_workspace" ]; then
- continue
- else
- PARAMS_FUNASR_LOCAL_WORKSPACE=$params_local_workspace
- mkdir -p $PARAMS_FUNASR_LOCAL_WORKSPACE
- fi
- fi
- if [ ! -d "$PARAMS_FUNASR_LOCAL_WORKSPACE" ]; then
- echo -e " ${RED}The local workspace(${PARAMS_FUNASR_LOCAL_WORKSPACE}) set does not exist, please setup again.${PLAIN}"
- else
- PARAMS_FUNASR_LOCAL_MODELS_DIR="${PARAMS_FUNASR_LOCAL_WORKSPACE}/models"
- echo -e " The local workspace path is ${GREEN}${PARAMS_FUNASR_LOCAL_WORKSPACE}${PLAIN} ."
- echo -e " The models will store in local path(${GREEN}${PARAMS_FUNASR_LOCAL_MODELS_DIR}${PLAIN}) during the run."
-
- break
- fi
- done
+ cd $CUR_DIR
echo
}
# Get a list of docker images and select them.
selectDockerImages(){
- echo -e "${UNDERLINE}${BOLD}[2/6]${PLAIN}"
+ echo -e "${UNDERLINE}${BOLD}[1/6]${PLAIN}"
echo -e " ${YELLOW}Getting the list of docker images, please wait a few seconds.${PLAIN}"
readDockerInfoFromUrl
echo
+ the_latest_docker_image=$PARAMS_DOCKER_IMAGE
+
echo -e " ${YELLOW}Please choose the Docker image.${PLAIN}"
menuSelection ${DOCKER_IMAGES[*]}
result=$?
- index=`expr $result - 1`
+ index=`expr ${result} - 1`
PARAMS_DOCKER_IMAGE=${DOCKER_IMAGES[${index}]}
echo -e " ${UNDERLINE}You have chosen the Docker image:${PLAIN} ${GREEN}${PARAMS_DOCKER_IMAGE}${PLAIN}"
- checkDockerExist
- result=$?
- result=`expr $result + 0`
- if [ ${result} -eq 50 ]; then
- return 50
+ if [ -z "$the_latest_docker_image" ] && [ -z "$PARAMS_FUNASR_DOCKER_ID" ]; then
+ result=0
+ else
+ # 0: DOCKER is not running
+ # 60: DOCKER_ID is empty
+ # 61: DOCKER_IMAGE is empty
+ # 62: DOCKER is running
+ # 63: DOCKER_ID and DOCKER_IMAGE are empty
+ checkDockerIdExist "install"
+ result=$?
+ result=`expr ${result} + 0`
+ if [ $result -eq 60 ]; then
+ result=0
+ elif [ $result -eq 61 ]; then
+ echo
+ echo -e " ${RED}Please run (${PLAIN}${GREEN}${SUDO_CMD} bash funasr-runtime-deploy-offline-cpu-zh.sh install${PLAIN}${RED}) to install Docker first.${PLAIN}"
+ elif [ $result -eq 62 ]; then
+ echo
+ echo -e " ${RED}Docker: ${PARAMS_DOCKER_IMAGE} ${PARAMS_FUNASR_DOCKER_ID} has been launched, please run (${PLAIN}${GREEN}${SUDO_CMD} bash funasr-runtime-deploy-offline-cpu-zh.sh remove${PLAIN}${RED}) to remove Docker first ant then install.${PLAIN}"
+ elif [ $result -eq 63 ]; then
+ result=0
+ fi
fi
+ echo
+}
+
+# Get a list of models and select them.
+selectModels(){
+ echo -e "${UNDERLINE}${BOLD}[2/6]${PLAIN}"
+ echo -e " ${YELLOW}Get a list of selectable models.${PLAIN}"
+ echo
+
+ selectAsrModels
+ selectVadModels
+ selectPuncModels
+
+ echo
+}
+
+selectAsrModels(){
+ echo -e " ${YELLOW}Please choose the ASR model.${PLAIN}"
+ menuSelection ${ASR_MODELS[*]}
+ result=$?
+ index=`expr ${result} - 1`
+
+ PARAMS_ASR_ID=${ASR_MODELS[${index}]}
+ echo -e " ${UNDERLINE}You have chosen the ASR model:${PLAIN} ${GREEN}${PARAMS_ASR_ID}${PLAIN}"
+ echo
+}
+
+selectVadModels(){
+ echo -e " ${YELLOW}Please choose the VAD model.${PLAIN}"
+ menuSelection ${VAD_MODELS[*]}
+ result=$?
+ index=`expr ${result} - 1`
+
+ PARAMS_VAD_ID=${VAD_MODELS[${index}]}
+ echo -e " ${UNDERLINE}You have chosen the VAD model:${PLAIN} ${GREEN}${PARAMS_VAD_ID}${PLAIN}"
+ echo
+}
+
+selectPuncModels(){
+ echo -e " ${YELLOW}Please choose the PUNC model.${PLAIN}"
+ menuSelection ${PUNC_MODELS[*]}
+ result=$?
+ index=`expr ${result} - 1`
+
+ PARAMS_PUNC_ID=${PUNC_MODELS[${index}]}
+ echo -e " ${UNDERLINE}You have chosen the PUNC model:${PLAIN} ${GREEN}${PARAMS_PUNC_ID}${PLAIN}"
echo
}
@@ -520,32 +619,99 @@
paramsFromDefault(){
initConfiguration
- echo -e "${UNDERLINE}${BOLD}[1-3/6]${PLAIN}"
- echo -e " ${YELLOW}Load parameters from ${DEFAULT_FUNASR_CONFIG_FILE}${PLAIN}"
+ echo -e " ${YELLOW}Load parameters from${PLAIN} ${GREEN}${DEFAULT_FUNASR_CONFIG_FILE}${PLAIN}"
echo
- PARAMS_FUNASR_LOCAL_WORKSPACE=`sed '/^PARAMS_FUNASR_LOCAL_WORKSPACE=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_FUNASR_SAMPLES_LOCAL_DIR=`sed '/^PARAMS_FUNASR_SAMPLES_LOCAL_DIR=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_FUNASR_SAMPLES_LOCAL_PATH=`sed '/^PARAMS_FUNASR_SAMPLES_LOCAL_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_FUNASR_LOCAL_MODELS_DIR=`sed '/^PARAMS_FUNASR_LOCAL_MODELS_DIR=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_FUNASR_CONFIG_PATH=`sed '/^PARAMS_FUNASR_CONFIG_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ funasr_local_workspace=`sed '/^PARAMS_FUNASR_LOCAL_WORKSPACE=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$funasr_local_workspace" ]; then
+ PARAMS_FUNASR_LOCAL_WORKSPACE=$funasr_local_workspace
+ fi
+ funasr_samples_local_dir=`sed '/^PARAMS_FUNASR_SAMPLES_LOCAL_DIR=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$funasr_samples_local_dir" ]; then
+ PARAMS_FUNASR_SAMPLES_LOCAL_DIR=$funasr_samples_local_dir
+ fi
+ funasr_samples_local_path=`sed '/^PARAMS_FUNASR_SAMPLES_LOCAL_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$funasr_samples_local_path" ]; then
+ PARAMS_FUNASR_SAMPLES_LOCAL_PATH=$funasr_samples_local_path
+ fi
+ funasr_local_models_dir=`sed '/^PARAMS_FUNASR_LOCAL_MODELS_DIR=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$funasr_local_models_dir" ]; then
+ PARAMS_FUNASR_LOCAL_MODELS_DIR=$funasr_local_models_dir
+ fi
- PARAMS_DOCKER_IMAGE=`sed '/^PARAMS_DOCKER_IMAGE=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_DOWNLOAD_MODEL_DIR=`sed '/^PARAMS_DOWNLOAD_MODEL_DIR=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ docker_image=`sed '/^PARAMS_DOCKER_IMAGE=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$docker_image" ]; then
+ PARAMS_DOCKER_IMAGE=$docker_image
+ fi
+ download_model_dir=`sed '/^PARAMS_DOWNLOAD_MODEL_DIR=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$download_model_dir" ]; then
+ PARAMS_DOWNLOAD_MODEL_DIR=$download_model_dir
+ fi
PARAMS_LOCAL_ASR_PATH=`sed '/^PARAMS_LOCAL_ASR_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_DOCKER_ASR_PATH=`sed '/^PARAMS_DOCKER_ASR_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_ASR_ID=`sed '/^PARAMS_ASR_ID=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_LOCAL_VAD_PATH=`sed '/^PARAMS_LOCAL_VAD_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_DOCKER_VAD_PATH=`sed '/^PARAMS_DOCKER_VAD_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_VAD_ID=`sed '/^PARAMS_VAD_ID=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_LOCAL_PUNC_PATH=`sed '/^PARAMS_LOCAL_PUNC_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_DOCKER_PUNC_PATH=`sed '/^PARAMS_DOCKER_PUNC_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_PUNC_ID=`sed '/^PARAMS_PUNC_ID=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_DOCKER_EXEC_PATH=`sed '/^PARAMS_DOCKER_EXEC_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_HOST_PORT=`sed '/^PARAMS_HOST_PORT=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_DOCKER_PORT=`sed '/^PARAMS_DOCKER_PORT=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_DECODER_THREAD_NUM=`sed '/^PARAMS_DECODER_THREAD_NUM=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- PARAMS_IO_THREAD_NUM=`sed '/^PARAMS_IO_THREAD_NUM=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$local_asr_path" ]; then
+ PARAMS_LOCAL_ASR_PATH=$local_asr_path
+ fi
+ docker_asr_path=`sed '/^PARAMS_DOCKER_ASR_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$docker_asr_path" ]; then
+ PARAMS_DOCKER_ASR_PATH=$docker_asr_path
+ fi
+ asr_id=`sed '/^PARAMS_ASR_ID=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$asr_id" ]; then
+ PARAMS_ASR_ID=$asr_id
+ fi
+ local_vad_path=`sed '/^PARAMS_LOCAL_VAD_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$local_vad_path" ]; then
+ PARAMS_LOCAL_VAD_PATH=$local_vad_path
+ fi
+ docker_vad_path=`sed '/^PARAMS_DOCKER_VAD_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$docker_vad_path" ]; then
+ PARAMS_DOCKER_VAD_PATH=$docker_vad_path
+ fi
+ vad_id=`sed '/^PARAMS_VAD_ID=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$vad_id" ]; then
+ PARAMS_VAD_ID=$vad_id
+ fi
+ local_punc_path=`sed '/^PARAMS_LOCAL_PUNC_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$local_punc_path" ]; then
+ PARAMS_LOCAL_PUNC_PATH=$local_punc_path
+ fi
+ docker_punc_path=`sed '/^PARAMS_DOCKER_PUNC_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$docker_punc_path" ]; then
+ PARAMS_DOCKER_PUNC_PATH=$docker_punc_path
+ fi
+ punc_id=`sed '/^PARAMS_PUNC_ID=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$punc_id" ]; then
+ PARAMS_PUNC_ID=$punc_id
+ fi
+ docker_exec_path=`sed '/^PARAMS_DOCKER_EXEC_PATH=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$docker_exec_path" ]; then
+ PARAMS_DOCKER_EXEC_PATH=$docker_exec_path
+ fi
+ host_port=`sed '/^PARAMS_HOST_PORT=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$host_port" ]; then
+ PARAMS_HOST_PORT=$host_port
+ fi
+ docker_port=`sed '/^PARAMS_DOCKER_PORT=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$docker_port" ]; then
+ PARAMS_DOCKER_PORT=$docker_port
+ fi
+ decode_thread_num=`sed '/^PARAMS_DECODER_THREAD_NUM=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$decode_thread_num" ]; then
+ PARAMS_DECODER_THREAD_NUM=$decode_thread_num
+ fi
+ io_thread_num=`sed '/^PARAMS_IO_THREAD_NUM=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$io_thread_num" ]; then
+ PARAMS_IO_THREAD_NUM=$io_thread_num
+ fi
+
+ ssl_flag=`sed '/^PARAMS_SSL_FLAG=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$ssl_flag" ]; then
+ PARAMS_SSL_FLAG=$ssl_flag
+ fi
+ docker_id=`sed '/^PARAMS_FUNASR_DOCKER_ID=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ ! -z "$docker_id" ]; then
+ PARAMS_FUNASR_DOCKER_ID=$docker_id
+ fi
}
saveParams(){
@@ -558,7 +724,6 @@
echo "PARAMS_FUNASR_SAMPLES_LOCAL_DIR=${PARAMS_FUNASR_SAMPLES_LOCAL_DIR}" >> $DEFAULT_FUNASR_CONFIG_FILE
echo "PARAMS_FUNASR_SAMPLES_LOCAL_PATH=${PARAMS_FUNASR_SAMPLES_LOCAL_PATH}" >> $DEFAULT_FUNASR_CONFIG_FILE
echo "PARAMS_FUNASR_LOCAL_MODELS_DIR=${PARAMS_FUNASR_LOCAL_MODELS_DIR}" >> $DEFAULT_FUNASR_CONFIG_FILE
- echo "PARAMS_FUNASR_CONFIG_PATH=${PARAMS_FUNASR_CONFIG_PATH}" >> $DEFAULT_FUNASR_CONFIG_FILE
echo "PARAMS_DOWNLOAD_MODEL_DIR=${PARAMS_DOWNLOAD_MODEL_DIR}" >> $DEFAULT_FUNASR_CONFIG_FILE
@@ -587,12 +752,19 @@
echo "PARAMS_DOCKER_PORT=${PARAMS_DOCKER_PORT}" >> $DEFAULT_FUNASR_CONFIG_FILE
echo "PARAMS_DECODER_THREAD_NUM=${PARAMS_DECODER_THREAD_NUM}" >> $DEFAULT_FUNASR_CONFIG_FILE
echo "PARAMS_IO_THREAD_NUM=${PARAMS_IO_THREAD_NUM}" >> $DEFAULT_FUNASR_CONFIG_FILE
+ echo "PARAMS_SSL_FLAG=${PARAMS_SSL_FLAG}" >> $DEFAULT_FUNASR_CONFIG_FILE
+ echo "PARAMS_FUNASR_DOCKER_ID=${PARAMS_FUNASR_DOCKER_ID}" >> $DEFAULT_FUNASR_CONFIG_FILE
+
+ serverConfigGeneration
+ echo "${daemon_server_config}" > $DEFAULT_FUNASR_SERVER_CONFIG_FILE
}
showAllParams(){
echo -e "${UNDERLINE}${BOLD}[4/6]${PLAIN}"
echo -e " ${YELLOW}Show parameters of FunASR server setting and confirm to run ...${PLAIN}"
echo
+
+ only_show_flag=$1
if [ ! -z "$PARAMS_DOCKER_IMAGE" ]; then
echo -e " The current Docker image is : ${GREEN}${PARAMS_DOCKER_IMAGE}${PLAIN}"
@@ -643,8 +815,19 @@
if [ ! -z "$PARAMS_FUNASR_SAMPLES_LOCAL_DIR" ]; then
echo -e " Sample code will be store in local : ${GREEN}${PARAMS_FUNASR_SAMPLES_LOCAL_DIR}${PLAIN}"
fi
+ if [ ! -z "$PARAMS_SSL_FLAG" ]; then
+ echo -e " The flag for the use of SSL : ${GREEN}${PARAMS_SSL_FLAG}${PLAIN}"
+ fi
+ if [ "$only_show_flag" = "only_show" ] && [ ! -z "$PARAMS_FUNASR_DOCKER_ID" ]; then
+ echo -e " The docker ID that already exists is : ${GREEN}${PARAMS_FUNASR_DOCKER_ID}${PLAIN}"
+ fi
echo
+
+ if [ "$only_show_flag" = "only_show" ]; then
+ return 0
+ fi
+
while true
do
params_confirm="y"
@@ -685,7 +868,7 @@
if [ $DOCKERINFOLEN -gt 30 ]; then
echo -e " ${YELLOW}Docker has installed.${PLAIN}"
else
- lowercase_osid=$(echo $OSID | tr '[A-Z]' '[a-z]')
+ lowercase_osid=$(echo ${OSID} | tr '[A-Z]' '[a-z]')
echo -e " ${YELLOW}Start install docker for ${lowercase_osid} ${PLAIN}"
DOCKER_INSTALL_CMD="curl -fsSL https://get.docker.com | bash -s docker --mirror Aliyun"
DOCKER_INSTALL_RUN_CMD=""
@@ -693,22 +876,22 @@
case "$lowercase_osid" in
ubuntu)
DOCKER_INSTALL_CMD="curl -fsSL https://test.docker.com -o test-docker.sh"
- DOCKER_INSTALL_RUN_CMD="sudo sh test-docker.sh"
+ DOCKER_INSTALL_RUN_CMD="${SUDO_CMD} sh test-docker.sh"
;;
centos)
DOCKER_INSTALL_CMD="curl -fsSL https://get.docker.com | bash -s docker --mirror Aliyun"
;;
debian)
DOCKER_INSTALL_CMD="curl -fsSL https://get.docker.com -o get-docker.sh"
- DOCKER_INSTALL_RUN_CMD="sudo sh get-docker.sh"
+ DOCKER_INSTALL_RUN_CMD="${SUDO_CMD} sh get-docker.sh"
;;
\"alios\")
DOCKER_INSTALL_CMD="curl -fsSL https://get.docker.com -o get-docker.sh"
- DOCKER_INSTALL_RUN_CMD="sudo sh get-docker.sh"
+ DOCKER_INSTALL_RUN_CMD="${SUDO_CMD} sh get-docker.sh"
;;
\"alinux\")
- DOCKER_INSTALL_CMD="sudo yum -y install dnf"
- DOCKER_INSTALL_RUN_CMD="sudo dnf -y install docker"
+ DOCKER_INSTALL_CMD="${SUDO_CMD} yum -y install dnf"
+ DOCKER_INSTALL_RUN_CMD="${SUDO_CMD} dnf -y install docker"
;;
*)
echo -e " ${RED}$lowercase_osid is not supported.${PLAIN}"
@@ -722,13 +905,13 @@
if [ ! -z "$DOCKER_INSTALL_RUN_CMD" ]; then
$DOCKER_INSTALL_RUN_CMD
fi
- sudo systemctl start docker
+ $SUDO_CMD systemctl start docker
- DOCKERINFO=$(sudo docker info | wc -l)
+ DOCKERINFO=$(${SUDO_CMD} docker info | wc -l)
DOCKERINFOLEN=`expr ${DOCKERINFO} + 0`
if [ $DOCKERINFOLEN -gt 30 ]; then
echo -e " ${GREEN}Docker install success, start docker server.${PLAIN}"
- sudo systemctl start docker
+ $SUDO_CMD systemctl start docker
else
echo -e " ${RED}Docker install failed!${PLAIN}"
exit 1
@@ -741,7 +924,7 @@
# Download docker image
echo -e " ${YELLOW}Pull docker image(${PARAMS_DOCKER_IMAGE})...${PLAIN}"
- sudo docker pull $PARAMS_DOCKER_IMAGE
+ ${SUDO_CMD} docker pull $PARAMS_DOCKER_IMAGE
echo
sleep 1
@@ -751,90 +934,59 @@
echo -e "${UNDERLINE}${BOLD}[6/6]${PLAIN}"
echo -e " ${YELLOW}Construct command and run docker ...${PLAIN}"
- RUN_CMD="sudo docker run"
- PORT_MAP=" -p ${PARAMS_HOST_PORT}:${PARAMS_DOCKER_PORT}"
- DIR_PARAMS=" --privileged=true"
- DIR_MAP_PARAMS=""
- if [ ! -z "$PARAMS_LOCAL_ASR_DIR" ]; then
- if [ -z "$DIR_MAP_PARAMS" ]; then
- DIR_MAP_PARAMS="${DIR_PARAMS} -v ${PARAMS_LOCAL_ASR_DIR}:${PARAMS_DOCKER_ASR_DIR}"
- else
- DIR_MAP_PARAMS="${DIR_MAP_PARAMS} -v ${PARAMS_LOCAL_ASR_DIR}:${PARAMS_DOCKER_ASR_DIR}"
+ start_flag=$1
+ if [ "$start_flag" = "install" ]; then
+ run_cmd="${SUDO_CMD} docker run"
+ port_map=" -p ${PARAMS_HOST_PORT}:${PARAMS_DOCKER_PORT}"
+ env_params=" --privileged=true"
+ dir_map_params=" -v ${DEFAULT_FUNASR_CONFIG_DIR}:/workspace/.config -v ${PARAMS_FUNASR_LOCAL_MODELS_DIR}:${PARAMS_DOWNLOAD_MODEL_DIR}"
+
+ serverConfigGeneration
+ env_params=" ${env_params} --env DAEMON_SERVER_CONFIG=${daemon_server_config}"
+
+ run_cmd="${run_cmd}${port_map}${dir_map_params}${env_params}"
+ run_cmd="${run_cmd} -it -d ${PARAMS_DOCKER_IMAGE}"
+ else
+ # 0: DOCKER is not running
+ # 60: DOCKER_ID is empty
+ # 61: DOCKER_IMAGE is empty
+ # 62: DOCKER is running
+ checkDockerIdExist $start_flag
+ result=$?
+ result=`expr ${result} + 0`
+ if [ $result -eq 60 ]; then
+ echo
+ echo -e " ${RED}Please run (${PLAIN}${GREEN}${SUDO_CMD} bash funasr-runtime-deploy-offline-cpu-zh.sh install${PLAIN}${RED}) to install Docker first.${PLAIN}"
+ return $result
+ elif [ $result -eq 61 ]; then
+ echo
+ echo -e " ${RED}Please run (${PLAIN}${GREEN}${SUDO_CMD} bash funasr-runtime-deploy-offline-cpu-zh.sh install${PLAIN}${RED}) to install Docker first.${PLAIN}"
+ return $result
+ elif [ $result -eq 62 ]; then
+ echo
+ echo -e " ${RED}Docker: ${PARAMS_DOCKER_IMAGE} ${PARAMS_FUNASR_DOCKER_ID} has been launched, please run (${PLAIN}${GREEN}${SUDO_CMD} bash funasr-runtime-deploy-offline-cpu-zh.sh stop${PLAIN}${RED}) to stop Docker first.${PLAIN}"
+ return $result
fi
- fi
- if [ ! -z "$PARAMS_LOCAL_VAD_DIR" ]; then
- if [ -z "$DIR_MAP_PARAMS" ]; then
- DIR_MAP_PARAMS="${DIR_PARAMS} -v ${PARAMS_LOCAL_VAD_DIR}:${PARAMS_DOCKER_VAD_DIR}"
- else
- DIR_MAP_PARAMS="${DIR_MAP_PARAMS} -v ${PARAMS_LOCAL_VAD_DIR}:${PARAMS_DOCKER_VAD_DIR}"
- fi
- fi
- if [ ! -z "$PARAMS_LOCAL_PUNC_DIR" ]; then
- if [ -z "$DIR_MAP_PARAMS" ]; then
- DIR_MAP_PARAMS="${DIR_PARAMS} -v ${PARAMS_LOCAL_PUNC_DIR}:${PARAMS_DOCKER_PUNC_DIR}"
- else
- DIR_MAP_PARAMS="${DIR_MAP_PARAMS} -v ${PARAMS_LOCAL_VAD_DIR}:${PARAMS_DOCKER_VAD_DIR}"
- fi
+
+ run_cmd="${SUDO_CMD} docker restart ${PARAMS_FUNASR_DOCKER_ID}"
fi
- EXEC_PARAMS="\"exec\":\"${PARAMS_DOCKER_EXEC_PATH}\""
- if [ ! -z "$PARAMS_ASR_ID" ]; then
- ASR_PARAMS="\"--model-dir\":\"${PARAMS_ASR_ID}\""
- else
- ASR_PARAMS="\"--model-dir\":\"${PARAMS_DOCKER_ASR_PATH}\""
- fi
- if [ ! -z "$PARAMS_VAD_ID" ]; then
- VAD_PARAMS="\"--vad-dir\":\"${PARAMS_VAD_ID}\""
- else
- VAD_PARAMS="\"--vad-dir\":\"${PARAMS_DOCKER_VAD_PATH}\""
- fi
- if [ ! -z "$PARAMS_PUNC_ID" ]; then
- PUNC_PARAMS="\"--punc-dir\":\"${PARAMS_PUNC_ID}\""
- else
- PUNC_PARAMS="\"--punc-dir\":\"${PARAMS_DOCKER_PUNC_PATH}\""
- fi
- DOWNLOAD_PARARMS="\"--download-model-dir\":\"${PARAMS_DOWNLOAD_MODEL_DIR}\""
- if [ -z "$PARAMS_DOWNLOAD_MODEL_DIR" ]; then
- MODEL_PARAMS="${ASR_PARAMS},${VAD_PARAMS},${PUNC_PARAMS}"
- else
- MODEL_PARAMS="${ASR_PARAMS},${VAD_PARAMS},${PUNC_PARAMS},${DOWNLOAD_PARARMS}"
- fi
+ rm -f ${DEFAULT_FUNASR_PROGRESS_TXT}
+ rm -f ${DEFAULT_FUNASR_SERVER_LOG}
- DECODER_PARAMS="\"--decoder-thread-num\":\"${PARAMS_DECODER_THREAD_NUM}\""
- IO_PARAMS="\"--io-thread-num\":\"${PARAMS_IO_THREAD_NUM}\""
- THREAD_PARAMS=${DECODER_PARAMS},${IO_PARAMS}
- PORT_PARAMS="\"--port\":\"${PARAMS_DOCKER_PORT}\""
- CRT_PATH="\"--certfile\":\"/workspace/FunASR/funasr/runtime/ssl_key/server.crt\""
- KEY_PATH="\"--keyfile\":\"/workspace/FunASR/funasr/runtime/ssl_key/server.key\""
-
- ENV_PARAMS=" -v /var/funasr:/workspace/.config"
- ENV_PARAMS=" ${ENV_PARAMS} --env DAEMON_SERVER_CONFIG={\"server\":[{${EXEC_PARAMS},${MODEL_PARAMS},${THREAD_PARAMS},${PORT_PARAMS},${CRT_PATH},${KEY_PATH}}]}"
-
- RUN_CMD="${RUN_CMD}${PORT_MAP}${DIR_MAP_PARAMS}${ENV_PARAMS}"
- RUN_CMD="${RUN_CMD} -it -d ${PARAMS_DOCKER_IMAGE}"
-
- # check Docker
- checkDockerExist
- result=$?
- result=`expr ${result} + 0`
- if [ ${result} -eq 50 ]; then
- return 50
- fi
-
- server_log="/var/funasr/server_console.log"
- rm -f ${PROGRESS_TXT}
- rm -f ${server_log}
-
- ${RUN_CMD}
+ $run_cmd
echo
echo -e " ${YELLOW}Loading models:${PLAIN}"
+
+ getDockerId
+ saveParams
# Hide the cursor, start draw progress.
printf "\e[?25l"
while true
do
- ServerProgress
+ serverProgress
result=$?
stage=`expr ${result} + 0`
if [ ${stage} -eq 0 ]; then
@@ -861,41 +1013,205 @@
echo -e " ${GREEN}The service has been started.${PLAIN}"
echo
- downloadSamples
+ deploySamples
echo -e " ${BOLD}The sample code is already stored in the ${PLAIN}(${GREEN}${PARAMS_FUNASR_SAMPLES_LOCAL_DIR}${PLAIN}) ."
- echo -e " ${BOLD}If you want to see an example of how to use the client, you can run ${PLAIN}${GREEN}sudo bash funasr-runtime-deploy-offline-cpu-zh.sh -c${PLAIN} ."
+ echo -e " ${BOLD}If you want to see an example of how to use the client, you can run ${PLAIN}${GREEN}${SUDO_CMD} bash funasr-runtime-deploy-offline-cpu-zh.sh client${PLAIN} ."
echo
}
-downloadSamples(){
- if [ ! -d $PARAMS_FUNASR_SAMPLES_LOCAL_DIR ]; then
- echo -e "${YELLOW}Downloading samples to $PARAMS_FUNASR_LOCAL_WORKSPACE ...${PLAIN}"
+daemon_server_config=""
+serverConfigGeneration(){
+ # params about models
+ if [ ! -z "$PARAMS_ASR_ID" ]; then
+ asr_params="\"--model-dir\":\"${PARAMS_ASR_ID}\""
+ else
+ if [ ! -z "$PARAMS_LOCAL_ASR_PATH" ]; then
+ dir_map_params="${dir_map_params} -v ${PARAMS_LOCAL_ASR_PATH}:${PARAMS_DOCKER_ASR_PATH}"
+ fi
+ asr_params="\"--model-dir\":\"${PARAMS_DOCKER_ASR_PATH}\""
+ fi
+ if [ ! -z "$PARAMS_VAD_ID" ]; then
+ vad_params="\"--vad-dir\":\"${PARAMS_VAD_ID}\""
+ else
+ if [ ! -z "$PARAMS_LOCAL_VAD_PATH" ]; then
+ dir_map_params="${dir_map_params} -v ${PARAMS_LOCAL_VAD_PATH}:${PARAMS_DOCKER_VAD_PATH}"
+ fi
+ vad_params="\"--vad-dir\":\"${PARAMS_DOCKER_VAD_PATH}\""
+ fi
+ if [ ! -z "$PARAMS_PUNC_ID" ]; then
+ punc_params="\"--punc-dir\":\"${PARAMS_PUNC_ID}\""
+ else
+ if [ ! -z "$PARAMS_LOCAL_PUNC_PATH" ]; then
+ dir_map_params="${dir_map_params} -v ${PARAMS_LOCAL_VAD_PATH}:${PARAMS_DOCKER_VAD_PATH}"
+ fi
+ punc_params="\"--punc-dir\":\"${PARAMS_DOCKER_PUNC_PATH}\""
+ fi
+ download_params="\"--download-model-dir\":\"${PARAMS_DOWNLOAD_MODEL_DIR}\""
+ model_params="${asr_params},${vad_params},${punc_params},${download_params}"
- DOWNLOAD_CMD="curl ${DEFAULT_SAMPLES_URL} -o ${PARAMS_FUNASR_SAMPLES_LOCAL_PATH}"
- UNTAR_CMD="tar -zxf ${PARAMS_FUNASR_SAMPLES_LOCAL_PATH} -C ${PARAMS_FUNASR_LOCAL_WORKSPACE}"
+ # params about thread_num
+ decoder_params="\"--decoder-thread-num\":\"${PARAMS_DECODER_THREAD_NUM}\""
+ io_params="\"--io-thread-num\":\"${PARAMS_IO_THREAD_NUM}\""
+ thread_params=${decoder_params},${io_params}
+
+ # params about port and ssl
+ port_params="\"--port\":\"${PARAMS_DOCKER_PORT}\""
+ if [ $PARAMS_SSL_FLAG -eq 0 ]; then
+ crt_path="\"--certfile\":\"\""
+ key_path="\"--keyfile\":\"\""
+ else
+ crt_path="\"--certfile\":\"/workspace/FunASR/funasr/runtime/ssl_key/server.crt\""
+ key_path="\"--keyfile\":\"/workspace/FunASR/funasr/runtime/ssl_key/server.key\""
+ fi
+
+ exec_params="\"exec\":\"${PARAMS_DOCKER_EXEC_PATH}\""
+
+ daemon_server_config="{\"server\":[{${exec_params},${model_params},${thread_params},${port_params},${crt_path},${key_path}}]}"
+}
+
+installPythonDependencyForPython(){
+ echo -e "${YELLOW}Install Python dependent environments ...${PLAIN}"
+
+ echo -e " Export dependency of Cpp sample."
+ pre_cmd="export LD_LIBRARY_PATH=${PARAMS_FUNASR_SAMPLES_LOCAL_DIR}/cpp/libs:\$LD_LIBRARY_PATH"
+ $pre_cmd
+ echo
+
+ echo -e " Install requirements of Python sample."
+ pre_cmd="pip3 install click>=8.0.4"
+ $pre_cmd
+ echo
+
+ pre_cmd="pip3 install -r ${PARAMS_FUNASR_SAMPLES_LOCAL_DIR}/python/requirements_client.txt"
+ echo -e " Run ${BLUE}${pre_cmd}${PLAIN}"
+ $pre_cmd
+ echo
+
+ lowercase_osid=$(echo ${OSID} | tr '[A-Z]' '[a-z]')
+ case "$lowercase_osid" in
+ ubuntu)
+ pre_cmd="${SUDO_CMD} apt-get install -y ffmpeg"
+ ;;
+ centos)
+ pre_cmd="${SUDO_CMD} yum install -y ffmpeg"
+ ;;
+ debian)
+ pre_cmd="${SUDO_CMD} apt-get install -y ffmpeg"
+ ;;
+ \"alios\")
+ pre_cmd="${SUDO_CMD} yum install -y ffmpeg"
+ ;;
+ \"alinux\")
+ pre_cmd="${SUDO_CMD} yum install -y ffmpeg"
+ ;;
+ *)
+ echo -e " ${RED}$lowercase_osid is not supported.${PLAIN}"
+ ;;
+ esac
+ echo -e " Run ${BLUE}${pre_cmd}${PLAIN}"
+ echo
+
+ pre_cmd="pip3 install ffmpeg-python"
+ echo -e " Run ${BLUE}${pre_cmd}${PLAIN}"
+ $pre_cmd
+ echo
+}
+
+deploySamples(){
+ if [ ! -d $PARAMS_FUNASR_SAMPLES_LOCAL_DIR ]; then
+ echo -e "${YELLOW}Downloading samples to${PLAIN} ${CYAN}${PARAMS_FUNASR_LOCAL_WORKSPACE}${PLAIN} ${YELLOW}...${PLAIN}"
+
+ download_cmd="curl ${DEFAULT_SAMPLES_URL} -o ${PARAMS_FUNASR_SAMPLES_LOCAL_PATH}"
+ untar_cmd="tar -zxf ${PARAMS_FUNASR_SAMPLES_LOCAL_PATH} -C ${PARAMS_FUNASR_LOCAL_WORKSPACE}"
if [ ! -f "$PARAMS_FUNASR_SAMPLES_LOCAL_PATH" ]; then
- $DOWNLOAD_CMD
+ $download_cmd
fi
- $UNTAR_CMD
- fi
-}
-
-checkDockerExist(){
- result=$(sudo docker ps | grep ${PARAMS_DOCKER_IMAGE} | wc -l)
- result=`expr ${result} + 0`
- if [ ${result} -ne 0 ]; then
+ $untar_cmd
echo
- echo -e " ${RED}Docker: ${PARAMS_DOCKER_IMAGE} has been launched, please run (${PLAIN}${GREEN}sudo bash funasr-runtime-deploy.sh -p${PLAIN}${RED}) to stop Docker first.${PLAIN}"
- return 50
+
+ installPythonDependencyForPython
+ echo
fi
}
-dockerExit(){
- echo -e " ${YELLOW}Stop docker(${PARAMS_DOCKER_IMAGE}) server ...${PLAIN}"
- sudo docker stop `sudo docker ps -a| grep ${PARAMS_DOCKER_IMAGE} | awk '{print $1}' `
+getDockerId(){
+ id=""
+ array=($(${SUDO_CMD} docker ps -a | grep ${PARAMS_DOCKER_IMAGE} | awk '{print $1}'))
+ len=${#array[@]}
+ if [ $len -ge 1 ]; then
+ # get the first id
+ id=$array
+ if [ ! -z "$id" ]; then
+ PARAMS_FUNASR_DOCKER_ID=$id
+ fi
+ fi
+}
+
+checkDockerImageExist(){
+ result=1
+ if [ -z "$PARAMS_DOCKER_IMAGE" ]; then
+ return 50
+ else
+ result=$(${SUDO_CMD} docker ps | grep ${PARAMS_DOCKER_IMAGE} | wc -l)
+ fi
+ result=`expr ${result} + 0`
+ echo "checkDockerImageExist result0: " $result
+ if [ $result -ne 0 ]; then
+ # found docker
+ return 51
+ else
+ return 0
+ fi
+}
+
+checkDockerIdExist(){
+ result=0
+ if [ -z "$PARAMS_FUNASR_DOCKER_ID" ]; then
+ if [ -z "$PARAMS_DOCKER_IMAGE" ]; then
+ return 63
+ else
+ return 60
+ fi
+ else
+ if [ -z "$PARAMS_DOCKER_IMAGE" ]; then
+ return 61
+ else
+ if [ "$1" = "install" ]; then
+ result=$(${SUDO_CMD} docker ps -a | grep ${PARAMS_DOCKER_IMAGE} | grep ${PARAMS_FUNASR_DOCKER_ID} | wc -l)
+ else
+ result=$(${SUDO_CMD} docker ps | grep ${PARAMS_DOCKER_IMAGE} | grep ${PARAMS_FUNASR_DOCKER_ID} | wc -l)
+ fi
+ fi
+ fi
+ result=`expr ${result} + 0`
+ if [ $result -eq 1 ]; then
+ # found docker
+ return 62
+ else
+ return 0
+ fi
+}
+
+dockerStop(){
+ if [ -z "$PARAMS_FUNASR_DOCKER_ID" ]; then
+ echo -e " ${RED}DOCKER_ID is empty, cannot stop docker.${PLAIN}"
+ else
+ echo -e " ${YELLOW}Stop docker(${PLAIN}${GREEN}${PARAMS_DOCKER_IMAGE} ${PARAMS_FUNASR_DOCKER_ID}${PLAIN}${YELLOW}) server ...${PLAIN}"
+ ${SUDO_CMD} docker stop ${PARAMS_FUNASR_DOCKER_ID}
+ fi
echo
- sleep 1
+}
+
+dockerRemove(){
+ if [ -z "$PARAMS_FUNASR_DOCKER_ID" ]; then
+ echo -e " ${RED}DOCKER_ID is empty, cannot remove docker.${PLAIN}"
+ else
+ echo -e " ${YELLOW}Remove docker(${PLAIN}${GREEN}${PARAMS_DOCKER_IMAGE} ${PARAMS_FUNASR_DOCKER_ID}${PLAIN}${YELLOW}) ...${PLAIN}"
+ ${SUDO_CMD} docker rm ${PARAMS_FUNASR_DOCKER_ID}
+ fi
+
+ echo
}
modelChange(){
@@ -903,13 +1219,16 @@
model_id=$2
local_flag=0
- if [ -d "$model_id" ]; then
+ relativePathToFullPath $model_id
+ if [ -d "$full_path" ]; then
local_flag=1
+ model_id=$full_path
else
local_flag=0
fi
+ full_path=""
- result=$(echo $model_type | grep "asr_model")
+ result=$(echo ${model_type} | grep "\-\-asr_model")
if [ "$result" != "" ]; then
if [ $local_flag -eq 0 ]; then
PARAMS_ASR_ID=$model_id
@@ -925,13 +1244,12 @@
else
model_name=$(basename "${PARAMS_LOCAL_ASR_PATH}")
PARAMS_LOCAL_ASR_DIR=$(dirname "${PARAMS_LOCAL_ASR_PATH}")
- middle=${PARAMS_LOCAL_ASR_DIR#*"${PARAMS_FUNASR_LOCAL_MODELS_DIR}"}
PARAMS_DOCKER_ASR_DIR=$PARAMS_DOWNLOAD_MODEL_DIR
- PARAMS_DOCKER_ASR_PATH=${PARAMS_DOCKER_ASR_DIR}/${middle}/${model_name}
+ PARAMS_DOCKER_ASR_PATH=${PARAMS_DOCKER_ASR_DIR}/${model_name}
fi
fi
fi
- result=$(echo ${model_type} | grep "vad_model")
+ result=$(echo ${model_type} | grep "\-\-vad_model")
if [ "$result" != "" ]; then
if [ $local_flag -eq 0 ]; then
PARAMS_VAD_ID=$model_id
@@ -947,13 +1265,12 @@
else
model_name=$(basename "${PARAMS_LOCAL_VAD_PATH}")
PARAMS_LOCAL_VAD_DIR=$(dirname "${PARAMS_LOCAL_VAD_PATH}")
- middle=${PARAMS_LOCAL_VAD_DIR#*"${PARAMS_FUNASR_LOCAL_MODELS_DIR}"}
PARAMS_DOCKER_VAD_DIR=$PARAMS_DOWNLOAD_MODEL_DIR
- PARAMS_DOCKER_VAD_PATH=${PARAMS_DOCKER_VAD_DIR}/${middle}/${model_name}
+ PARAMS_DOCKER_VAD_PATH=${PARAMS_DOCKER_VAD_DIR}/${model_name}
fi
fi
fi
- result=$(echo $model_type | grep "punc_model")
+ result=$(echo ${model_type} | grep "\-\-punc_model")
if [ "$result" != "" ]; then
if [ $local_flag -eq 0 ]; then
PARAMS_PUNC_ID=$model_id
@@ -964,9 +1281,8 @@
else
model_name=$(basename "${PARAMS_LOCAL_PUNC_PATH}")
PARAMS_LOCAL_PUNC_DIR=$(dirname "${PARAMS_LOCAL_PUNC_PATH}")
- middle=${PARAMS_LOCAL_PUNC_DIR#*"${PARAMS_FUNASR_LOCAL_MODELS_DIR}"}
PARAMS_DOCKER_PUNC_DIR=$PARAMS_DOWNLOAD_MODEL_DIR
- PARAMS_DOCKER_PUNC_PATH=${PARAMS_DOCKER_PUNC_DIR}/${middle}/${model_name}
+ PARAMS_DOCKER_PUNC_PATH=${PARAMS_DOCKER_PUNC_DIR}/${model_name}
fi
fi
}
@@ -978,11 +1294,11 @@
if [ -z "$val"]; then
num=`expr ${val} + 0`
if [ $num -ge 1 ] && [ $num -le 1024 ]; then
- result=$(echo ${type} | grep "decode_thread_num")
+ result=$(echo ${type} | grep "\-\-decode_thread_num")
if [ "$result" != "" ]; then
PARAMS_DECODER_THREAD_NUM=$num
fi
- result=$(echo ${type} | grep "io_thread_num")
+ result=$(echo ${type} | grep "\-\-io_thread_num")
if [ "$result" != "" ]; then
PARAMS_IO_THREAD_NUM=$num
fi
@@ -1012,17 +1328,16 @@
sampleClientRun(){
echo -e "${YELLOW}Will download sample tools for the client to show how speech recognition works.${PLAIN}"
- DOWNLOAD_CMD="curl ${DEFAULT_SAMPLES_URL} -o ${PARAMS_FUNASR_SAMPLES_LOCAL_PATH}"
- UNTAR_CMD="tar -zxf ${PARAMS_FUNASR_SAMPLES_LOCAL_PATH} ${PARAMS_FUNASR_LOCAL_WORKSPACE}"
+ download_cmd="curl ${DEFAULT_SAMPLES_URL} -o ${PARAMS_FUNASR_SAMPLES_LOCAL_PATH}"
+ untar_cmd="tar -zxf ${PARAMS_FUNASR_SAMPLES_LOCAL_PATH} -C ${PARAMS_FUNASR_LOCAL_WORKSPACE}"
if [ ! -f "$PARAMS_FUNASR_SAMPLES_LOCAL_PATH" ]; then
- $DOWNLOAD_CMD
+ $download_cmd
fi
if [ -f "$PARAMS_FUNASR_SAMPLES_LOCAL_PATH" ]; then
- $UNTAR_CMD
+ $untar_cmd
fi
if [ -d "$PARAMS_FUNASR_SAMPLES_LOCAL_DIR" ]; then
-
echo -e " Please select the client you want to run."
menuSelection ${SAMPLE_CLIENTS[*]}
result=$?
@@ -1030,51 +1345,50 @@
lang=${SAMPLE_CLIENTS[${index}]}
echo
- SERVER_IP="127.0.0.1"
- read -p " Please enter the IP of server, default($SERVER_IP): " SERVER_IP
- if [ -z "$SERVER_IP" ]; then
- SERVER_IP="127.0.0.1"
+ server_ip="127.0.0.1"
+ echo -e " Please enter the IP of server, default(${CYAN}${server_ip}${PLAIN}): \c"
+ read server_ip
+ if [ -z "$server_ip" ]; then
+ server_ip="127.0.0.1"
fi
- HOST_PORT=`sed '/^PARAMS_HOST_PORT=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- if [ -z "$HOST_PORT" ]; then
- HOST_PORT="10095"
+ host_port=`sed '/^PARAMS_HOST_PORT=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ -z "$host_port" ]; then
+ host_port="10095"
fi
- read -p " Please enter the port of server, default($HOST_PORT): " HOST_PORT
- if [ -z "$HOST_PORT" ]; then
- HOST_PORT=`sed '/^PARAMS_HOST_PORT=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
- if [ -z "$HOST_PORT" ]; then
- HOST_PORT="10095"
+ echo -e " Please enter the port of server, default(${CYAN}${host_port}${PLAIN}): \c"
+ read host_port
+ if [ -z "$host_port" ]; then
+ host_port=`sed '/^PARAMS_HOST_PORT=/!d;s/.*=//' ${DEFAULT_FUNASR_CONFIG_FILE}`
+ if [ -z "$host_port" ]; then
+ host_port="10095"
fi
fi
- WAV_PATH="${PARAMS_FUNASR_LOCAL_WORKSPACE}/funasr_samples/audio/asr_example.wav"
- read -p " Please enter the audio path, default(${WAV_PATH}): " WAV_PATH
- if [ -z "$WAV_PATH" ]; then
- WAV_PATH="${PARAMS_FUNASR_LOCAL_WORKSPACE}/funasr_samples/audio/asr_example.wav"
+ wav_path="${PARAMS_FUNASR_SAMPLES_LOCAL_DIR}/audio/asr_example.wav"
+ echo -e " Please enter the audio path, default(${CYAN}${wav_path}${PLAIN}): \c"
+ read WAV_PATH
+ if [ -z "$wav_path" ]; then
+ wav_path="${PARAMS_FUNASR_SAMPLES_LOCAL_DIR}/audio/asr_example.wav"
fi
echo
- PRE_CMD=鈥濃��
+ pre_cmd=鈥濃��
case "$lang" in
Linux_Cpp)
- PRE_CMD="export LD_LIBRARY_PATH=${PARAMS_FUNASR_LOCAL_WORKSPACE}/funasr_samples/cpp/libs:\$LD_LIBRARY_PATH"
- CLIENT_EXEC="${PARAMS_FUNASR_LOCAL_WORKSPACE}/funasr_samples/cpp/funasr-wss-client"
- RUN_CMD="${CLIENT_EXEC} --server-ip ${SERVER_IP} --port ${HOST_PORT} --wav-path ${WAV_PATH}"
- echo -e " Run ${BLUE}${PRE_CMD}${PLAIN}"
- $PRE_CMD
- echo
+ client_exec="${PARAMS_FUNASR_SAMPLES_LOCAL_DIR}/cpp/funasr-wss-client"
+ run_cmd="${client_exec} --server-ip ${server_ip} --port ${host_port} --wav-path ${wav_path}"
;;
Python)
- CLIENT_EXEC="${PARAMS_FUNASR_LOCAL_WORKSPACE}/funasr_samples/python/wss_client_asr.py"
- RUN_CMD="python3 ${CLIENT_EXEC} --host ${SERVER_IP} --port ${HOST_PORT} --mode offline --audio_in ${WAV_PATH} --send_without_sleep --output_dir ${PARAMS_FUNASR_LOCAL_WORKSPACE}/funasr_samples/python"
- PRE_CMD="pip3 install click>=8.0.4"
- echo -e " Run ${BLUE}${PRE_CMD}${PLAIN}"
- $PRE_CMD
+ client_exec="${PARAMS_FUNASR_SAMPLES_LOCAL_DIR}/python/wss_client_asr.py"
+ run_cmd="python3 ${client_exec} --host ${server_ip} --port ${host_port} --mode offline --audio_in ${wav_path} --send_without_sleep --output_dir ${PARAMS_FUNASR_SAMPLES_LOCAL_DIR}/python"
+ pre_cmd="pip3 install click>=8.0.4"
+ echo -e " Run ${BLUE}${pre_cmd}${PLAIN}"
+ $pre_cmd
echo
- PRE_CMD="pip3 install -r ${PARAMS_FUNASR_LOCAL_WORKSPACE}/funasr_samples/python/requirements_client.txt"
- echo -e " Run ${BLUE}${PRE_CMD}${PLAIN}"
- $PRE_CMD
+ pre_cmd="pip3 install -r ${PARAMS_FUNASR_SAMPLES_LOCAL_DIR}/python/requirements_client.txt"
+ echo -e " Run ${BLUE}${pre_cmd}${PLAIN}"
+ $pre_cmd
echo
;;
*)
@@ -1082,27 +1396,28 @@
;;
esac
- echo -e " Run ${BLUE}${RUN_CMD}${PLAIN}"
- $RUN_CMD
+ echo -e " Run ${BLUE}${run_cmd}${PLAIN}"
+ $run_cmd
echo
- echo -e " If failed, you can try (${GREEN}${RUN_CMD}${PLAIN}) in your Shell."
+ echo -e " If failed, you can try (${GREEN}${run_cmd}${PLAIN}) in your Shell."
echo
fi
}
paramsConfigure(){
initConfiguration
- setupLocalWorkspaceDir
initParameters
selectDockerImages
+ selectModels
result=$?
result=`expr ${result} + 0`
- if [ ${result} -eq 50 ]; then
- return 50
+ if [ $result -ne 0 ]; then
+ return $result
fi
setupHostPort
complementParameters
+ return 0
}
# Display Help info
@@ -1110,17 +1425,22 @@
echo -e "${UNDERLINE}Usage${PLAIN}:"
echo -e " $0 [OPTIONAL FLAGS]"
echo
- echo -e "funasr-runtime-deploy-offline-cpu.sh - a Bash script to install&run FunASR docker."
+ echo -e "funasr-runtime-deploy-offline-cpu-zh.sh - a Bash script to install&run FunASR docker."
echo
echo -e "${UNDERLINE}Options${PLAIN}:"
echo -e " ${BOLD}-i, install, --install${PLAIN} Install and run FunASR docker."
+ echo -e " install [--workspace] <workspace in local>"
+ echo -e " install [--ssl] <0: close SSL; 1: open SSL, default:1>"
echo -e " ${BOLD}-s, start , --start${PLAIN} Run FunASR docker with configuration that has already been set."
echo -e " ${BOLD}-p, stop , --stop${PLAIN} Stop FunASR docker."
+ echo -e " ${BOLD}-m, remove , --remove${PLAIN} Remove FunASR docker installed."
echo -e " ${BOLD}-r, restart, --restart${PLAIN} Restart FunASR docker."
echo -e " ${BOLD}-u, update , --update${PLAIN} Update parameters that has already been set."
- echo -e " --update [asr_model | vad_model | punc_model] <model_id or local model path>"
- echo -e " --update [host_port | docker_port] <port number>"
- echo -e " --update [decode_thread_num | io_thread_num] <the number of threads>"
+ echo -e " update [--workspace] <workspace in local>"
+ echo -e " update [--asr_model | --vad_model | --punc_model] <model_id or local model path>"
+ echo -e " update [--host_port | --docker_port] <port number>"
+ echo -e " update [--decode_thread_num | io_thread_num] <the number of threads>"
+ echo -e " update [--ssl] <0: close SSL; 1: open SSL, default:1>"
echo -e " ${BOLD}-c, client , --client${PLAIN} Get a client example to show how to initiate speech recognition."
echo -e " ${BOLD}-o, show , --show${PLAIN} Displays all parameters that have been set."
echo -e " ${BOLD}-v, version, --version${PLAIN} Display current script version."
@@ -1144,7 +1464,14 @@
stage=$result
else
if [ "$stage" = "--workspace" ]; then
- DEFAULT_FUNASR_LOCAL_WORKSPACE=$val
+ relativePathToFullPath $val
+ PARAMS_FUNASR_LOCAL_WORKSPACE=$full_path
+ full_path=""
+ if [ ! -z "$PARAMS_FUNASR_LOCAL_WORKSPACE" ]; then
+ mkdir -p $PARAMS_FUNASR_LOCAL_WORKSPACE
+ fi
+ elif [ "$stage" = "--ssl" ]; then
+ PARAMS_SSL_FLAG=`expr ${val} + 0`
fi
fi
done
@@ -1155,21 +1482,21 @@
OSID=$(grep ^ID= /etc/os-release | cut -d= -f2)
OSVER=$(lsb_release -cs)
OSNUM=$(grep -oE "[0-9.]+" /etc/issue)
-CPUNUM=$(cat /proc/cpuinfo |grep "processor"|wc -l)
-DOCKERINFO=$(sudo docker info | wc -l)
+CPUNUM=$(cat /proc/cpuinfo | grep "processor"|wc -l)
+DOCKERINFO=$(${SUDO_CMD} docker info | wc -l)
DOCKERINFOLEN=`expr ${DOCKERINFO} + 0`
# PARAMS
# The workspace for FunASR in local
PARAMS_FUNASR_LOCAL_WORKSPACE=$DEFAULT_FUNASR_LOCAL_WORKSPACE
# The dir stored sample code in local
-PARAMS_FUNASR_SAMPLES_LOCAL_DIR=${PARAMS_FUNASR_LOCAL_WORKSPACE}/${DEFAULT_SAMPLES_NAME}
+PARAMS_FUNASR_SAMPLES_LOCAL_DIR=${PARAMS_FUNASR_LOCAL_WORKSPACE}/${DEFAULT_SAMPLES_DIR}
# The path of sample code in local
PARAMS_FUNASR_SAMPLES_LOCAL_PATH=${PARAMS_FUNASR_LOCAL_WORKSPACE}/${DEFAULT_SAMPLES_NAME}.tar.gz
# The dir stored models in local
PARAMS_FUNASR_LOCAL_MODELS_DIR="${PARAMS_FUNASR_LOCAL_WORKSPACE}/models"
-# The path of configuration in local
-PARAMS_FUNASR_CONFIG_PATH="${PARAMS_FUNASR_LOCAL_WORKSPACE}/config"
+# The id of started docker
+PARAMS_FUNASR_DOCKER_ID=""
# The server excutor in local
PARAMS_DOCKER_EXEC_PATH=$DEFAULT_DOCKER_EXEC_PATH
@@ -1218,6 +1545,7 @@
PARAMS_DOCKER_PORT="10095"
PARAMS_DECODER_THREAD_NUM="32"
PARAMS_IO_THREAD_NUM="8"
+PARAMS_SSL_FLAG=1
echo -e "#############################################################"
@@ -1236,51 +1564,102 @@
case "$1" in
install|-i|--install)
rootNess
+ paramsFromDefault
parseInput $@
paramsConfigure
result=$?
result=`expr ${result} + 0`
- if [ ${result} -ne 50 ]; then
- showAllParams
+ if [ $result -eq 0 ]; then
+ showAllParams "install"
installFunasrDocker
- dockerRun
+ dockerRun "install"
result=$?
- stage=`expr ${result} + 0`
- if [ $stage -eq 98 ]; then
- dockerExit
- dockerRun
- fi
+
+ try_count=1
+ while true
+ do
+ stage=`expr ${result} + 0`
+ if [ $try_count -ge 10 ]; then
+ break
+ else
+ # 98: cannot find progress from Docker
+ if [ $stage -eq 98 ]; then
+ dockerStop
+ dockerRun "start"
+ result=$?
+ let try_count=try_count+1
+ else
+ break
+ fi
+ fi
+ done
fi
;;
start|-s|--start)
rootNess
paramsFromDefault
- showAllParams
- dockerRun
+ showAllParams "only_show"
+ dockerRun "start"
result=$?
- stage=`expr ${result} + 0`
- if [ $stage -eq 98 ]; then
- dockerExit
- dockerRun
- fi
+
+ try_count=1
+ while true
+ do
+ stage=`expr ${result} + 0`
+ if [ $try_count -ge 10 ]; then
+ break
+ else
+ # 98: cannot find progress from Docker
+ if [ $stage -eq 98 ]; then
+ dockerStop
+ dockerRun "start"
+ result=$?
+ let try_count=try_count+1
+ else
+ break
+ fi
+ fi
+ done
;;
restart|-r|--restart)
rootNess
paramsFromDefault
- showAllParams
- dockerExit
- dockerRun
+ showAllParams "only_show"
+ dockerStop
+ dockerRun "start"
result=$?
- stage=`expr ${result} + 0`
- if [ $stage -eq 98 ]; then
- dockerExit
- dockerRun
- fi
+
+ try_count=1
+ while true
+ do
+ stage=`expr ${result} + 0`
+ if [ $try_count -ge 10 ]; then
+ break
+ else
+ # 98: cannot find progress from Docker
+ if [ $stage -eq 98 ]; then
+ dockerStop
+ dockerRun "start"
+ result=$?
+ let try_count=try_count+1
+ else
+ break
+ fi
+ fi
+ done
;;
stop|-p|--stop)
rootNess
paramsFromDefault
- dockerExit
+ dockerStop
+ ;;
+ remove|-m|--remove)
+ rootNess
+ paramsFromDefault
+ dockerStop
+ dockerRemove
+ rm -f ${DEFAULT_FUNASR_CONFIG_FILE}
+ rm -f ${DEFAULT_FUNASR_SERVER_CONFIG_FILE}
;;
update|-u|--update)
rootNess
@@ -1289,40 +1668,69 @@
if [ $# -eq 3 ]; then
type=$2
val=$3
- if [ "$type" = "asr_model" ] || [ "$type" = "vad_model" ] || [ "$type" = "punc_model" ]; then
+ if [ "$type" = "--asr_model" ] || [ "$type" = "--vad_model" ] || [ "$type" = "--punc_model" ]; then
modelChange $type $val
- elif [ "$type" = "decode_thread_num" ] || [ "$type" = "io_thread_num" ]; then
+ elif [ "$type" = "--decode_thread_num" ] || [ "$type" = "--io_thread_num" ]; then
threadNumChange $type $val
- elif [ "$type" = "host_port" ] || [ "$type" = "docker_port" ]; then
+ elif [ "$type" = "--host_port" ] || [ "$type" = "--docker_port" ]; then
portChange $type $val
+ elif [ "$type" = "--workspace" ]; then
+ relativePathToFullPath $val
+ PARAMS_FUNASR_LOCAL_WORKSPACE=$full_path
+ if [ ! -z "$PARAMS_FUNASR_LOCAL_WORKSPACE" ]; then
+ mkdir -p $PARAMS_FUNASR_LOCAL_WORKSPACE
+ fi
+ elif [ "$type" = "--ssl" ]; then
+ switch=`expr ${val} + 0`
+ if [ $switch -eq 0 ]; then
+ PARAMS_SSL_FLAG=0
+ else
+ PARAMS_SSL_FLAG=1
+ fi
else
- setupLocalWorkspaceDir
+ displayHelp
fi
else
- setupLocalWorkspaceDir
+ displayHelp
fi
- complementParameters
initParameters
- showAllParams
- dockerExit
- dockerRun
+ complementParameters
+ showAllParams "install"
+ dockerStop
+ dockerRun "start"
result=$?
- stage=`expr ${result} + 0`
- if [ $stage -eq 98 ]; then
- dockerExit
- dockerRun
- fi
+
+ try_count=1
+ while true
+ do
+ stage=`expr ${result} + 0`
+ if [ $try_count -ge 10 ]; then
+ break
+ else
+ # 98: cannot find progress from Docker
+ # 60: DOCKER_ID is empty
+ if [ $stage -eq 98 ] || [ $stage -eq 60 ]; then
+ dockerStop
+ dockerRun "start"
+ result=$?
+ let try_count=try_count+1
+ else
+ break
+ fi
+ fi
+ done
;;
client|-c|--client)
rootNess
paramsFromDefault
+ parseInput $@
sampleClientRun
;;
show|-o|--show)
rootNess
paramsFromDefault
- showAllParams
+ showAllParams "only_show"
;;
*)
displayHelp
--
Gitblit v1.9.1