mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 18:24:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix data prepare to just use train text by uid

Browse Source
This commit is contained in:
PingFeng Luo 2021-12-03 11:25:24 +08:00
parent bc0b6eed5c
commit 00b5ac5815
1 changed files with 29 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
egs/aishell/ASR/prepare.sh
View File

@ -111,63 +111,67 @@ if [ $stage -le 4 ] && [ $stop_stage -ge 4 ]; then
./local/compute_fbank_musan.py
fi
lang_phone_dir=data/lang_phone
lang_char_dir=data/lang_char
if [ $stage -le 5 ] && [ $stop_stage -ge 5 ]; then
log "Stage 5: Prepare phone based lang"
lang_dir=data/lang_phone
mkdir -p $lang_dir
mkdir -p $lang_phone_dir
(echo '!SIL SIL'; echo '<SPOKEN_NOISE> SPN'; echo '<UNK> SPN'; ) |
cat - $dl_dir/aishell/resource_aishell/lexicon.txt |
sort | uniq > $lang_dir/lexicon.txt
sort | uniq > $lang_phone_dir/lexicon.txt
./local/generate_unique_lexicon.py --lang-dir $lang_dir
./local/generate_unique_lexicon.py --lang-dir $lang_phone_dir
if [ ! -f data/lang_phone/L_disambig.pt ]; then
if [ ! -f $lang_phone_dir/L_disambig.pt ]; then
./local/prepare_lang.py
fi
# Train a bigram P for MMI training
if [ ! -f $lang_dir/transcript_words.txt ]; then
if [ ! -f $lang_phone_dir/transcript_words.txt ]; then
log "Generate data to train phone based bigram P"
aishell_text=aishell/data_aishell/transcript/aishell_transcript_v0.8.txt
cat ${dl_dir}/${aishell_text} | cut -d " " -f 2- > $lang_dir/transcript_words.txt
train_uids=$(find data/aishell/data_aishell/wav/train -name "*.wav" | sed 's/\.wav//g' | awk -F '/' '{print $NF}')
aishell_text=$dl_dir/aishell/data_aishell/transcript/aishell_transcript_v0.8.txt
for uid in ${train_uids}; do
awk -v uid=$uid 'index($1, uid)' $aishell_text | cut -d " " -f 2-
done > $lang_phone_dir/transcript_words.txt
fi
if [ ! -f $lang_dir/transcript_tokens.txt ]; then
if [ ! -f $lang_phone_dir/transcript_tokens.txt ]; then
./local/convert_transcript_words_to_tokens.py \
--lexicon $lang_dir/uniq_lexicon.txt \
--transcript $lang_dir/transcript_words.txt \
--lexicon $lang_phone_dir/uniq_lexicon.txt \
--transcript $lang_phone_dir/transcript_words.txt \
--oov "<UNK>" \
> $lang_dir/transcript_tokens.txt
> $lang_phone_dir/transcript_tokens.txt
fi
if [ ! -f $lang_dir/P.arpa ]; then
if [ ! -f $lang_phone_dir/P.arpa ]; then
./shared/make_kn_lm.py \
-ngram-order 2 \
-text $lang_dir/transcript_tokens.txt \
-lm $lang_dir/P.arpa
-text $lang_phone_dir/transcript_tokens.txt \
-lm $lang_phone_dir/P.arpa
fi
if [ ! -f $lang_dir/P.fst.txt ]; then
if [ ! -f $lang_phone_dir/P.fst.txt ]; then
python3 -m kaldilm \
--read-symbol-table="$lang_dir/tokens.txt" \
--read-symbol-table="$lang_phone_dir/tokens.txt" \
--disambig-symbol='#0' \
--max-order=2 \
$lang_dir/P.arpa > $lang_dir/P.fst.txt
$lang_phone_dir/P.arpa > $lang_phone_dir/P.fst.txt
fi
fi
if [ $stage -le 6 ] && [ $stop_stage -ge 6 ]; then
log "Stage 6: Prepare char based lang"
mkdir -p data/lang_char
mkdir -p $lang_char_dir
# We reuse words.txt from phone based lexicon
# so that the two can share G.pt later.
cp data/lang_phone/words.txt data/lang_char
cp $lang_phone_dir/words.txt $lang_char_dir
cat $dl_dir/aishell/data_aishell/transcript/aishell_transcript_v0.8.txt |
cut -d " " -f 2- | sed -e 's/[ \t\r\n]*//g' > data/lang_char/text
cut -d " " -f 2- | sed -e 's/[ \t\r\n]*//g' > $lang_char_dir/text
if [ ! -f data/lang_char/L_disambig.pt ]; then
if [ ! -f $lang_char_dir/L_disambig.pt ]; then
./local/prepare_char.py
fi
fi
@ -181,7 +185,7 @@ if [ $stage -le 7 ] && [ $stop_stage -ge 7 ]; then
if [ ! -f data/lm/G_3_gram.fst.txt ]; then
# It is used in building HLG
python3 -m kaldilm \
--read-symbol-table="data/lang_phone/words.txt" \
--read-symbol-table="$lang_phone_dir/words.txt" \
--disambig-symbol='#0' \
--max-order=3 \
$dl_dir/lm/3-gram.unpruned.arpa > data/lm/G_3_gram.fst.txt
@ -190,6 +194,6 @@ fi
if [ $stage -le 8 ] && [ $stop_stage -ge 8 ]; then
log "Stage 8: Compile HLG"
./local/compile_hlg.py --lang-dir data/lang_phone
./local/compile_hlg.py --lang-dir data/lang_char
./local/compile_hlg.py --lang-dir $lang_phone_dir
./local/compile_hlg.py --lang-dir $lang_char_dir
fi