4a-build-vosk-model.sh

#!/bin/bash
# Collect all files of the adapted model (work folder) and build a Vosk-ASR compatible model
set -e
echo ---------------------------COLLECT-------------------------------
ORG_MODEL="model"
WORK_DIR="work"
TARGET_DIR="adapted_model"
if [ -n "$1" ]; then
	TARGET_DIR="$1"
fi
if [ -d "${TARGET_DIR}" ]; then
	echo "Removing old folder: ${TARGET_DIR} ..."
	rm -r "${TARGET_DIR}"
fi
mkdir -p "${TARGET_DIR}/am"
mkdir -p "${TARGET_DIR}/conf"
mkdir -p "${TARGET_DIR}/graph/phones"
mkdir -p "${TARGET_DIR}/ivector"
# collect
echo "Collecting files ..."
cp $WORK_DIR/exp/adapt/final.mdl "${TARGET_DIR}/am/final.mdl"
cp $WORK_DIR/conf/mfcc_hires.conf "${TARGET_DIR}/conf/mfcc.conf"
cp $WORK_DIR/exp/adapt/graph/HCLG.fst "${TARGET_DIR}/graph/HCLG.fst"
cp $WORK_DIR/exp/adapt/graph/num_pdfs "${TARGET_DIR}/graph/num_pdfs"
cp $WORK_DIR/exp/adapt/graph/num_pdfs "${TARGET_DIR}/graph/num_pdfs"
cp $WORK_DIR/exp/adapt/graph/phones.txt "${TARGET_DIR}/graph/phones.txt"
cp $WORK_DIR/exp/adapt/graph/words.txt "${TARGET_DIR}/graph/words.txt"
cp $WORK_DIR/exp/adapt/graph/phones/word_boundary.int "${TARGET_DIR}/graph/phones/word_boundary.int"
cp $WORK_DIR/exp/extractor/* "${TARGET_DIR}/ivector/"
rm "${TARGET_DIR}/ivector/splice_opts"
cp $WORK_DIR/conf/online_cmvn.conf "${TARGET_DIR}/ivector/online_cmvn.conf"
# copy from original model (this is kind of a default file that should probably be tweaked further)
cp $ORG_MODEL/conf/model.conf "${TARGET_DIR}/conf/model.conf"
cp $ORG_MODEL/ivectors_test_hires/conf/splice.conf "${TARGET_DIR}/ivector/splice.conf"
echo "DONE"
echo "Check folder: ${TARGET_DIR}"