mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 10:02:22 +00:00
28 lines
812 B
Bash
28 lines
812 B
Bash
wav_dir=$1
|
|
wav_files=$(ls $wav_dir/*.wav)
|
|
# wav_files=$(echo $wav_files | cut -d " " -f 1)
|
|
# if wav_files is empty, then exit
|
|
if [ -z "$wav_files" ]; then
|
|
exit 1
|
|
fi
|
|
label_file=$2
|
|
model_path=local/sherpa-onnx-paraformer-zh-2023-09-14
|
|
|
|
if [ ! -d $model_path ]; then
|
|
pip install sherpa-onnx
|
|
wget -nc https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models/sherpa-onnx-paraformer-zh-2023-09-14.tar.bz2
|
|
tar xvf sherpa-onnx-paraformer-zh-2023-09-14.tar.bz2 -C local
|
|
fi
|
|
|
|
python3 local/offline-decode-files.py \
|
|
--tokens=$model_path/tokens.txt \
|
|
--paraformer=$model_path/model.int8.onnx \
|
|
--num-threads=2 \
|
|
--decoding-method=greedy_search \
|
|
--debug=false \
|
|
--sample-rate=24000 \
|
|
--log-dir $wav_dir \
|
|
--feature-dim=80 \
|
|
--label $label_file \
|
|
$wav_files
|