From 8d19c056d76b88abebeb0902cf4cd98ecb62de7a Mon Sep 17 00:00:00 2001
From: Yang Zhou <goat.zhou@qq.com>
Date: Thu, 12 May 2022 19:55:17 +0800
Subject: [PATCH] add custom_streaming_asr

---
 demos/custom_streaming_asr/README.md          |  1 +
 demos/custom_streaming_asr/README_cn.md       |  0
 demos/custom_streaming_asr/path.sh            |  2 ++
 demos/custom_streaming_asr/setup_docker.sh    |  1 +
 .../custom_streaming_asr/websocket_client.sh  | 18 ++++++++++
 .../custom_streaming_asr/websocket_server.sh  | 33 +++++++++++++++++++
 6 files changed, 55 insertions(+)
 create mode 100644 demos/custom_streaming_asr/README.md
 create mode 100644 demos/custom_streaming_asr/README_cn.md
 create mode 100644 demos/custom_streaming_asr/path.sh
 create mode 100644 demos/custom_streaming_asr/setup_docker.sh
 create mode 100755 demos/custom_streaming_asr/websocket_client.sh
 create mode 100755 demos/custom_streaming_asr/websocket_server.sh

diff --git a/demos/custom_streaming_asr/README.md b/demos/custom_streaming_asr/README.md
new file mode 100644
index 00000000..16df311c
--- /dev/null
+++ b/demos/custom_streaming_asr/README.md
@@ -0,0 +1 @@
+run setup_docker.sh first and then run websocket_server.sh
diff --git a/demos/custom_streaming_asr/README_cn.md b/demos/custom_streaming_asr/README_cn.md
new file mode 100644
index 00000000..e69de29b
diff --git a/demos/custom_streaming_asr/path.sh b/demos/custom_streaming_asr/path.sh
new file mode 100644
index 00000000..47462324
--- /dev/null
+++ b/demos/custom_streaming_asr/path.sh
@@ -0,0 +1,2 @@
+export LD_LIBRARY_PATH=$PWD/resource/lib
+export PATH=$PATH:$PWD/resource/bin
diff --git a/demos/custom_streaming_asr/setup_docker.sh b/demos/custom_streaming_asr/setup_docker.sh
new file mode 100644
index 00000000..329a75db
--- /dev/null
+++ b/demos/custom_streaming_asr/setup_docker.sh
@@ -0,0 +1 @@
+sudo nvidia-docker run --privileged  --net=host --ipc=host -it --rm -v $PWD:/paddle --name=paddle_demo_docker registry.baidubce.com/paddlepaddle/paddle:2.2.2 /bin/bash
diff --git a/demos/custom_streaming_asr/websocket_client.sh b/demos/custom_streaming_asr/websocket_client.sh
new file mode 100755
index 00000000..ede076ca
--- /dev/null
+++ b/demos/custom_streaming_asr/websocket_client.sh
@@ -0,0 +1,18 @@
+#!/bin/bash
+set +x
+set -e
+
+. path.sh
+# input
+data=$PWD/data
+
+# output
+wav_scp=wav.scp
+
+export GLOG_logtostderr=1
+
+# websocket client
+websocket_client_main \
+    --wav_rspecifier=scp:$data/$wav_scp \
+    --streaming_chunk=0.36 \
+    --port=8881
diff --git a/demos/custom_streaming_asr/websocket_server.sh b/demos/custom_streaming_asr/websocket_server.sh
new file mode 100755
index 00000000..041c345b
--- /dev/null
+++ b/demos/custom_streaming_asr/websocket_server.sh
@@ -0,0 +1,33 @@
+#!/bin/bash
+set +x
+set -e
+
+export GLOG_logtostderr=1
+
+. path.sh
+#test websocket server 
+
+model_dir=./resource/model
+graph_dir=./resource/graph
+cmvn=./data/cmvn.ark
+
+
+#paddle_asr_online/resource.tar.gz
+if [ ! -f $cmvn ]; then
+    wget -c https://paddlespeech.bj.bcebos.com/s2t/paddle_asr_online/resource.tar.gz
+    tar xzfv resource.tar.gz
+    ln -s ./resource/data .
+fi
+
+websocket_server_main \
+    --cmvn_file=$cmvn \
+    --streaming_chunk=0.1 \
+    --use_fbank=true \
+    --model_path=$model_dir/avg_10.jit.pdmodel \
+    --param_path=$model_dir/avg_10.jit.pdiparams \
+    --model_cache_shapes="5-1-2048,5-1-2048" \
+    --model_output_names=softmax_0.tmp_0,tmp_5,concat_0.tmp_0,concat_1.tmp_0 \
+    --word_symbol_table=$graph_dir/words.txt \
+    --graph_path=$graph_dir/TLG.fst --max_active=7500 \
+    --port=8881 \
+    --acoustic_scale=12