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.
46 lines
979 B
46 lines
979 B
#! /usr/bin/env bash
|
|
|
|
mkdir -p data
|
|
|
|
TARGET_DIR=${MAIN_ROOT}/examples/dataset
|
|
mkdir -p ${TARGET_DIR}
|
|
|
|
# download data, generate manifests
|
|
PYTHONPATH=.:$PYTHONPATH python3 ${TARGET_DIR}/aishell/aishell.py \
|
|
--manifest_prefix="data/manifest" \
|
|
--target_dir="${TARGET_DIR}/aishell"
|
|
|
|
if [ $? -ne 0 ]; then
|
|
echo "Prepare Aishell failed. Terminated."
|
|
exit 1
|
|
fi
|
|
|
|
|
|
# build vocabulary
|
|
python3 ${MAIN_ROOT}/utils/build_vocab.py \
|
|
--count_threshold=0 \
|
|
--vocab_path="data/vocab.txt" \
|
|
--manifest_paths "data/manifest.train" "data/manifest.dev"
|
|
|
|
if [ $? -ne 0 ]; then
|
|
echo "Build vocabulary failed. Terminated."
|
|
exit 1
|
|
fi
|
|
|
|
|
|
# compute mean and stddev for normalizer
|
|
python3 ${MAIN_ROOT}/utils/compute_mean_std.py \
|
|
--manifest_path="data/manifest.train" \
|
|
--num_samples=2000 \
|
|
--specgram_type="linear" \
|
|
--output_path="data/mean_std.npz"
|
|
|
|
if [ $? -ne 0 ]; then
|
|
echo "Compute mean and stddev failed. Terminated."
|
|
exit 1
|
|
fi
|
|
|
|
|
|
echo "Aishell data preparation done."
|
|
exit 0
|