You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
2.7 KiB
112 lines
2.7 KiB
3 years ago
|
#!/bin/bash
|
||
|
|
|
||
|
|
stage=-1
|
||
|
|
stop_stage=100
|
||
|
|
|
||
|
|
# bpemode (unigram or bpe)
|
||
|
|
nbpe=8000
|
||
|
|
bpemode=unigram
|
||
|
|
bpeprefix="data/bpe_${bpemode}_${nbpe}"
|
||
|
|
DATA_DIR=
|
||
|
|
|
||
|
|
|
||
|
|
source ${MAIN_ROOT}/utils/parse_options.sh
|
||
|
|
|
||
|
|
|
||
|
|
mkdir -p data
|
||
|
|
TARGET_DIR=${MAIN_ROOT}/examples/dataset
|
||
|
|
mkdir -p ${TARGET_DIR}
|
||
|
|
|
||
|
|
if [ ! -d ${SOURCE_DIR} ]; then
|
||
|
|
echo "Error: Dataset is not avaiable. Please download and unzip the dataset"
|
||
|
|
echo "Download Link: https://pan.baidu.com/s/18L-59wgeS96WkObISrytQQ Passwd: bva0"
|
||
|
|
echo "The tree of the directory should be:"
|
||
|
|
echo "."
|
||
|
|
echo "|-- En-Zh"
|
||
|
|
echo "|-- test-segment"
|
||
|
|
echo " |-- tst2010"
|
||
|
|
echo " |-- ..."
|
||
|
|
echo "|-- train-split"
|
||
|
|
echo " |-- train-segment"
|
||
|
|
echo "|-- README.md"
|
||
|
|
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ ${stage} -le -1 ] && [ ${stop_stage} -ge -1 ]; then
|
||
|
|
# generate manifests
|
||
|
|
python3 ${TARGET_DIR}/ted_en_zh/ted_en_zh.py \
|
||
|
|
--manifest_prefix="data/manifest" \
|
||
|
|
--src_dir="${DATA_DIR}"
|
||
|
|
|
||
|
|
echo "Complete raw data pre-process."
|
||
|
|
fi
|
||
|
|
|
||
|
|
|
||
|
|
if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
|
||
|
|
# build vocabulary
|
||
|
|
python3 ${MAIN_ROOT}/utils/build_vocab.py \
|
||
|
|
--unit_type "spm" \
|
||
|
|
--spm_vocab_size=${nbpe} \
|
||
|
|
--spm_mode ${bpemode} \
|
||
|
|
--spm_model_prefix ${bpeprefix} \
|
||
|
|
--vocab_path="data/vocab.txt" \
|
||
|
|
--text_keys 'text' 'text1' \
|
||
|
|
--manifest_paths="data/manifest.train.raw"
|
||
|
|
|
||
|
|
|
||
|
|
if [ $? -ne 0 ]; then
|
||
|
|
echo "Build vocabulary failed. Terminated."
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
|
||
|
|
if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
|
||
|
|
# compute mean and stddev for normalizer
|
||
|
|
num_workers=$(nproc)
|
||
|
|
python3 ${MAIN_ROOT}/utils/compute_mean_std.py \
|
||
|
|
--manifest_path="data/manifest.train.raw" \
|
||
|
|
--num_samples=-1 \
|
||
|
|
--specgram_type="fbank" \
|
||
|
|
--feat_dim=80 \
|
||
|
|
--delta_delta=false \
|
||
|
|
--sample_rate=16000 \
|
||
|
|
--stride_ms=10.0 \
|
||
|
|
--window_ms=25.0 \
|
||
|
|
--use_dB_normalization=False \
|
||
|
|
--num_workers=${num_workers} \
|
||
|
|
--output_path="data/mean_std.json"
|
||
|
|
|
||
|
|
if [ $? -ne 0 ]; then
|
||
|
|
echo "Compute mean and stddev failed. Terminated."
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
|
||
|
|
if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
|
||
|
|
# format manifest with tokenids, vocab size
|
||
|
|
for set in train dev test; do
|
||
|
|
{
|
||
|
|
python3 ${MAIN_ROOT}/utils/format_triplet_data.py \
|
||
|
|
--feat_type "raw" \
|
||
|
|
--cmvn_path "data/mean_std.json" \
|
||
|
|
--unit_type "spm" \
|
||
|
|
--spm_model_prefix ${bpeprefix} \
|
||
|
|
--vocab_path="data/vocab.txt" \
|
||
|
|
--manifest_path="data/manifest.${set}.raw" \
|
||
|
|
--output_path="data/manifest.${set}"
|
||
|
|
|
||
|
|
if [ $? -ne 0 ]; then
|
||
|
|
echo "Formt mnaifest failed. Terminated."
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}&
|
||
|
|
done
|
||
|
|
wait
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "Ted En-Zh Data preparation done."
|
||
|
|
exit 0
|