imporove scripts

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 \
+if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
-compute_fbank_main \
+    cmvn_json2kaldi_main \
-    --num_bins 80 \
+        --json_file  $model_dir/mean_std.json \
-    --cmvn_file=$exp/cmvn.ark \
+        --cmvn_write_path $exp/cmvn.ark \
-    --streaming_chunk=36 \
+        --binary=false
-    --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 "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

speechx/examples/u2pp_ol/wenetspeech/run.sh

@@ -4,8 +4,8 @@ set -e
 . path.sh
 
 nj=40
-stage=0
+stage=-1
-stop_stage=5
+stop_stage=100
 
 . utils/parse_options.sh