From 2b80397cd3ee26281f2f98e17a653084cf477b83 Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Thu, 27 Oct 2022 08:09:18 +0000
Subject: [PATCH] imporove scripts

---
 .../u2pp_ol/wenetspeech/local/feat.sh         | 45 ++++++++++++-------
 speechx/examples/u2pp_ol/wenetspeech/run.sh   |  4 +-
 2 files changed, 30 insertions(+), 19 deletions(-)

diff --git a/speechx/examples/u2pp_ol/wenetspeech/local/feat.sh b/speechx/examples/u2pp_ol/wenetspeech/local/feat.sh
index 5a3188783..e181951e3 100755
--- a/speechx/examples/u2pp_ol/wenetspeech/local/feat.sh
+++ b/speechx/examples/u2pp_ol/wenetspeech/local/feat.sh
@@ -3,29 +3,40 @@ set -e
 
 . path.sh
 
+nj=20
+stage=-1
+stop_stage=100
+
+. utils/parse_options.sh
+
 data=data
 exp=exp
-nj=20
 mkdir -p $exp
+
 ckpt_dir=./data/model
 model_dir=$ckpt_dir/asr1_chunk_conformer_u2pp_wenetspeech_static_1.3.0.model/
 aishell_wav_scp=aishell_test.scp
 
-cmvn_json2kaldi_main \
-    --json_file  $model_dir/mean_std.json \
-    --cmvn_write_path $exp/cmvn.ark \
-    --binary=false
-
-echo "convert json cmvn to kaldi ark."
-
-./local/split_data.sh $data $data/$aishell_wav_scp $aishell_wav_scp $nj
 
-utils/run.pl JOB=1:$nj $data/split${nj}/JOB/feat.log \
-compute_fbank_main \
-    --num_bins 80 \
-    --cmvn_file=$exp/cmvn.ark \
-    --streaming_chunk=36 \
-    --wav_rspecifier=scp:$data/split${nj}/JOB/${aishell_wav_scp} \
-    --feature_wspecifier=ark,scp:$data/split${nj}/JOB/fbank.ark,$data/split${nj}/JOB/fbank.scp
+if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
+    cmvn_json2kaldi_main \
+        --json_file  $model_dir/mean_std.json \
+        --cmvn_write_path $exp/cmvn.ark \
+        --binary=false
+    
+    echo "convert json cmvn to kaldi ark."
+fi
 
-echo "compute fbank feature."
+if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
+    ./local/split_data.sh $data $data/$aishell_wav_scp $aishell_wav_scp $nj
+    
+    utils/run.pl JOB=1:$nj $data/split${nj}/JOB/feat.log \
+    compute_fbank_main \
+        --num_bins 80 \
+        --cmvn_file=$exp/cmvn.ark \
+        --streaming_chunk=36 \
+        --wav_rspecifier=scp:$data/split${nj}/JOB/${aishell_wav_scp} \
+        --feature_wspecifier=ark,scp:$data/split${nj}/JOB/fbank.ark,$data/split${nj}/JOB/fbank.scp
+    
+    echo "compute fbank feature."
+fi
diff --git a/speechx/examples/u2pp_ol/wenetspeech/run.sh b/speechx/examples/u2pp_ol/wenetspeech/run.sh
index 6de555845..4659ca0f7 100755
--- a/speechx/examples/u2pp_ol/wenetspeech/run.sh
+++ b/speechx/examples/u2pp_ol/wenetspeech/run.sh
@@ -4,8 +4,8 @@ set -e
 . path.sh
 
 nj=40
-stage=0
-stop_stage=5
+stage=-1
+stop_stage=100
 
 . utils/parse_options.sh