2018-03-28 10:31:33 +02:00
|
|
|
import sys
|
2019-01-24 09:38:28 +01:00
|
|
|
import os
|
2019-01-27 01:34:04 +01:00
|
|
|
os.chdir(r'C:\Users\Aki\source\repos\acoustic_model\acoustic_model')
|
2019-01-24 09:38:28 +01:00
|
|
|
|
2018-03-28 10:31:33 +02:00
|
|
|
import tempfile
|
2019-02-03 00:34:35 +01:00
|
|
|
import shutil
|
2019-02-03 13:54:37 +01:00
|
|
|
import glob
|
2019-01-27 23:52:33 +01:00
|
|
|
import time
|
2019-01-24 09:38:28 +01:00
|
|
|
|
2019-01-28 12:34:20 +01:00
|
|
|
import numpy as np
|
|
|
|
import pandas as pd
|
2018-03-26 20:50:14 +02:00
|
|
|
|
2019-01-24 09:38:28 +01:00
|
|
|
import fame_functions
|
2019-02-03 00:34:35 +01:00
|
|
|
from phoneset import fame_ipa, fame_asr
|
2019-01-24 09:38:28 +01:00
|
|
|
import defaultfiles as default
|
|
|
|
sys.path.append(default.toolbox_dir)
|
2019-01-27 01:34:04 +01:00
|
|
|
import file_handling as fh
|
|
|
|
from htk import pyhtk
|
2018-04-02 01:07:50 +02:00
|
|
|
|
2018-03-28 10:31:33 +02:00
|
|
|
|
|
|
|
## ======================= user define =======================
|
2018-04-25 09:07:46 +02:00
|
|
|
# procedure
|
2019-02-03 13:54:37 +01:00
|
|
|
make_lexicon = 0
|
2019-02-04 13:46:27 +01:00
|
|
|
make_label = 0 # it takes roughly 4800 sec on Surface pro 2.
|
2019-03-03 02:05:37 +01:00
|
|
|
make_mlf = 0
|
2019-02-04 13:46:27 +01:00
|
|
|
extract_features = 0
|
2019-01-29 21:52:11 +01:00
|
|
|
flat_start = 0
|
2019-02-04 20:32:12 +01:00
|
|
|
train_model_without_sp = 0
|
2019-03-07 22:16:50 +01:00
|
|
|
add_sp = 0
|
|
|
|
train_model_with_re_aligned_mlf = 1
|
2019-03-03 02:05:37 +01:00
|
|
|
train_triphone = 0
|
2019-02-04 20:32:12 +01:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
|
|
|
|
|
|
|
|
# pre-defined values.
|
|
|
|
dataset_list = ['devel', 'test', 'train']
|
2019-03-05 00:11:38 +01:00
|
|
|
feature_size = 39
|
2019-03-07 22:16:50 +01:00
|
|
|
improvement_threshold = 0.3
|
2019-03-05 00:11:38 +01:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
hmmdefs_name = 'hmmdefs'
|
2019-03-03 02:05:37 +01:00
|
|
|
proto_name = 'proto'
|
2019-02-04 13:46:27 +01:00
|
|
|
|
|
|
|
lexicon_asr = os.path.join(default.fame_dir, 'lexicon', 'lex.asr')
|
|
|
|
lexicon_oov = os.path.join(default.fame_dir, 'lexicon', 'lex.oov')
|
2018-03-28 10:31:33 +02:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
config_dir = os.path.join(default.htk_dir, 'config')
|
2019-03-03 02:05:37 +01:00
|
|
|
|
2019-02-04 20:32:12 +01:00
|
|
|
sil_hed = os.path.join(config_dir, 'sil.hed')
|
|
|
|
prototype = os.path.join(config_dir, proto_name)
|
2018-03-28 10:31:33 +02:00
|
|
|
|
2019-03-07 22:16:50 +01:00
|
|
|
model_dir = os.path.join(default.htk_dir, 'model')
|
|
|
|
model0_dir = os.path.join(model_dir, 'hmm0')
|
|
|
|
model1_dir = os.path.join(model_dir, 'hmm1')
|
|
|
|
model1sp_dir = os.path.join(model_dir, 'hmm1sp')
|
|
|
|
model1sp2_dir = os.path.join(model_dir, 'hmm1sp2')
|
2019-02-03 00:34:35 +01:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
# directories / files to be made.
|
|
|
|
lexicon_dir = os.path.join(default.htk_dir, 'lexicon')
|
|
|
|
lexicon_htk_asr = os.path.join(lexicon_dir, 'lex.htk_asr')
|
|
|
|
lexicon_htk_oov = os.path.join(lexicon_dir, 'lex.htk_oov')
|
|
|
|
lexicon_htk = os.path.join(lexicon_dir, 'lex.htk')
|
2019-01-24 09:38:28 +01:00
|
|
|
|
|
|
|
feature_dir = os.path.join(default.htk_dir, 'mfc')
|
2019-03-03 02:05:37 +01:00
|
|
|
fh.make_new_directory(feature_dir, existing_dir='leave')
|
2019-01-24 09:38:28 +01:00
|
|
|
tmp_dir = os.path.join(default.htk_dir, 'tmp')
|
2019-03-03 02:05:37 +01:00
|
|
|
fh.make_new_directory(tmp_dir, existing_dir='leave')
|
2019-02-03 00:34:35 +01:00
|
|
|
label_dir = os.path.join(default.htk_dir, 'label')
|
2019-03-03 02:05:37 +01:00
|
|
|
fh.make_new_directory(label_dir, existing_dir='leave')
|
2019-02-03 00:34:35 +01:00
|
|
|
|
2019-03-05 00:11:38 +01:00
|
|
|
|
2019-02-04 20:32:12 +01:00
|
|
|
## training
|
|
|
|
hcompv_scp_train = os.path.join(tmp_dir, 'train.scp')
|
|
|
|
mlf_file_train = os.path.join(label_dir, 'train_phone.mlf')
|
2019-02-14 00:21:28 +01:00
|
|
|
mlf_file_train_aligned = os.path.join(label_dir, 'train_phone_aligned.mlf')
|
2019-02-04 20:32:12 +01:00
|
|
|
|
2019-03-05 00:11:38 +01:00
|
|
|
## testing
|
|
|
|
htk_stimmen_dir = os.path.join(default.htk_dir, 'stimmen')
|
|
|
|
|
|
|
|
|
2019-01-29 21:52:11 +01:00
|
|
|
## ======================= make lexicon for HTK =======================
|
|
|
|
if make_lexicon:
|
2019-02-04 13:46:27 +01:00
|
|
|
timer_start = time.time()
|
|
|
|
print('==== making lexicon for HTK ====')
|
2019-01-29 21:52:11 +01:00
|
|
|
|
|
|
|
# convert each lexicon from fame_asr phoneset to fame_htk phoneset.
|
2019-02-04 13:46:27 +01:00
|
|
|
print('>>> converting each lexicon from fame_asr phoneset to fame_htk phoneset...')
|
2019-01-29 21:52:11 +01:00
|
|
|
fame_functions.lexicon_asr2htk(lexicon_asr, lexicon_htk_asr)
|
|
|
|
fame_functions.lexicon_asr2htk(lexicon_oov, lexicon_htk_oov)
|
2018-04-25 09:07:46 +02:00
|
|
|
|
|
|
|
# combine lexicon
|
2019-02-04 13:46:27 +01:00
|
|
|
print('>>> combining lexicon files into one lexicon...')
|
2018-04-25 09:07:46 +02:00
|
|
|
# pronunciations which is not found in lex.asr are generated using G2P and listed in lex.oov.
|
|
|
|
# therefore there is no overlap between lex_asr and lex_oov.
|
2019-01-29 21:52:11 +01:00
|
|
|
fame_functions.combine_lexicon(lexicon_htk_asr, lexicon_htk_oov, lexicon_htk)
|
|
|
|
|
2019-03-03 02:05:37 +01:00
|
|
|
## fixing the lexicon for HTK.
|
2019-01-29 21:52:11 +01:00
|
|
|
# (1) Replace all tabs with single space;
|
|
|
|
# (2) Put a '\' before any dictionary entry beginning with single quote
|
2019-03-03 02:05:37 +01:00
|
|
|
# http://electroblaze.blogspot.nl/2013/03/understanding-htk-error-messages.html
|
2019-02-14 00:21:28 +01:00
|
|
|
print('>>> fixing the lexicon...')
|
|
|
|
fame_functions.fix_lexicon(lexicon_htk)
|
2019-02-04 13:46:27 +01:00
|
|
|
print("elapsed time: {}".format(time.time() - timer_start))
|
2018-04-02 01:07:50 +02:00
|
|
|
|
|
|
|
|
2019-03-03 02:05:37 +01:00
|
|
|
## intialize the instance for HTK.
|
2019-03-05 00:11:38 +01:00
|
|
|
chtk = pyhtk.HTK(config_dir, fame_asr.phoneset_htk, lexicon_htk, feature_size)
|
2019-03-03 02:05:37 +01:00
|
|
|
|
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
## ======================= make label files =======================
|
|
|
|
if make_label:
|
2018-04-25 09:07:46 +02:00
|
|
|
for dataset in dataset_list:
|
2019-02-03 00:34:35 +01:00
|
|
|
timer_start = time.time()
|
2019-02-04 13:46:27 +01:00
|
|
|
print("==== making label files on dataset {}".format(dataset))
|
2018-04-25 09:07:46 +02:00
|
|
|
|
2019-02-03 00:34:35 +01:00
|
|
|
script_list = os.path.join(default.fame_dir, 'data', dataset, 'text')
|
2019-02-04 13:46:27 +01:00
|
|
|
wav_dir_ = os.path.join(default.fame_dir, 'fame', 'wav', dataset)
|
|
|
|
label_dir_ = os.path.join(label_dir, dataset)
|
|
|
|
dictionary_file = os.path.join(label_dir_, 'temp.dic')
|
2019-03-03 02:05:37 +01:00
|
|
|
fh.make_new_directory(label_dir_, existing_dir='leave')
|
2018-04-25 09:07:46 +02:00
|
|
|
|
|
|
|
# list of scripts
|
|
|
|
with open(script_list, "rt", encoding="utf-8") as fin:
|
2019-02-03 00:34:35 +01:00
|
|
|
scripts = fin.read().split('\n')
|
|
|
|
|
|
|
|
for line in scripts:
|
|
|
|
# sample line:
|
|
|
|
# sp0457m_test_1968_plakkenfryslanterhorne_2168 en dan begjinne je natuerlik
|
|
|
|
filename_ = line.split(' ')[0]
|
|
|
|
filename = '_'.join(filename_.split('_')[1:])
|
|
|
|
sentence = ' '.join(line.split(' ')[1:])
|
2019-02-03 13:54:37 +01:00
|
|
|
sentence_htk = fame_functions.word2htk(sentence)
|
2019-02-03 00:34:35 +01:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
wav_file = os.path.join(wav_dir_, filename + '.wav')
|
2019-03-03 02:05:37 +01:00
|
|
|
if os.path.exists(wav_file) and chtk.can_be_ascii(sentence_htk) == 0:
|
|
|
|
if chtk.get_number_of_missing_words(
|
|
|
|
sentence_htk, dictionary_file) == 0:
|
2019-02-03 13:54:37 +01:00
|
|
|
# when the file name is too long, HDMan command does not work.
|
|
|
|
# therefore first temporary dictionary_file is made, then renamed.
|
2019-02-04 13:46:27 +01:00
|
|
|
shutil.move(dictionary_file, os.path.join(label_dir_, filename + '.dic'))
|
|
|
|
|
|
|
|
label_file = os.path.join(label_dir_, filename + '.lab')
|
2019-03-03 02:05:37 +01:00
|
|
|
chtk.create_label_file(sentence_htk, label_file)
|
2019-02-03 13:54:37 +01:00
|
|
|
else:
|
|
|
|
os.remove(dictionary_file)
|
2019-03-03 02:05:37 +01:00
|
|
|
|
2019-02-03 00:34:35 +01:00
|
|
|
print("elapsed time: {}".format(time.time() - timer_start))
|
2019-02-03 13:54:37 +01:00
|
|
|
|
2018-04-25 09:07:46 +02:00
|
|
|
|
2019-03-03 02:05:37 +01:00
|
|
|
## ======================= make master label files =======================
|
|
|
|
if make_mlf:
|
2019-02-04 13:46:27 +01:00
|
|
|
timer_start = time.time()
|
2019-03-03 02:05:37 +01:00
|
|
|
print("==== making master label files ====")
|
2019-02-04 13:46:27 +01:00
|
|
|
|
2019-03-03 02:05:37 +01:00
|
|
|
# train_2002_gongfansaken_10347.lab is empty. should be removed.
|
|
|
|
empty_lab_file = os.path.join(label_dir, 'train', 'train_2002_gongfansaken_10347.lab')
|
|
|
|
empty_dic_file = empty_lab_file.replace('.lab', '.dic')
|
|
|
|
|
|
|
|
if os.path.exists(empty_lab_file):
|
|
|
|
os.remove(empty_lab_file)
|
|
|
|
if os.path.exists(empty_dic_file):
|
|
|
|
os.remove(empty_dic_file)
|
2019-02-03 13:54:37 +01:00
|
|
|
|
|
|
|
for dataset in dataset_list:
|
2019-03-03 02:05:37 +01:00
|
|
|
#wav_dir_ = os.path.join(default.fame_dir, 'fame', 'wav', dataset)
|
2019-02-04 13:46:27 +01:00
|
|
|
feature_dir_ = os.path.join(feature_dir, dataset)
|
|
|
|
label_dir_ = os.path.join(label_dir, dataset)
|
|
|
|
mlf_word = os.path.join(label_dir, dataset + '_word.mlf')
|
|
|
|
mlf_phone = os.path.join(label_dir, dataset + '_phone.mlf')
|
|
|
|
|
2019-03-03 02:05:37 +01:00
|
|
|
print(">>> generating a word level mlf file for {}...".format(dataset))
|
|
|
|
chtk.label2mlf(label_dir_, mlf_word)
|
|
|
|
print(">>> generating a phone level mlf file for {}...".format(dataset))
|
|
|
|
chtk.mlf_word2phone(mlf_phone, mlf_word)
|
|
|
|
|
|
|
|
print("elapsed time: {}".format(time.time() - timer_start))
|
2019-02-03 13:54:37 +01:00
|
|
|
|
2018-04-25 09:07:46 +02:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
## ======================= extract features =======================
|
|
|
|
if extract_features:
|
|
|
|
for dataset in dataset_list:
|
|
|
|
timer_start = time.time()
|
|
|
|
print('==== extract features on dataset {} ===='.format(dataset))
|
2018-04-25 09:07:46 +02:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
wav_dir_ = os.path.join(default.fame_dir, 'fame', 'wav', dataset)
|
|
|
|
label_dir_ = os.path.join(label_dir, dataset)
|
|
|
|
feature_dir_ = os.path.join(feature_dir, dataset)
|
2019-03-03 02:05:37 +01:00
|
|
|
fh.make_new_directory(feature_dir_, existing_dir='delete')
|
2018-04-25 09:07:46 +02:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
# a script file for HCopy
|
|
|
|
print(">>> making a script file for HCopy...")
|
|
|
|
hcopy_scp = tempfile.NamedTemporaryFile(mode='w', delete=False)
|
|
|
|
hcopy_scp.close()
|
2018-04-25 09:07:46 +02:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
# get a list of features (hcopy.scp)
|
|
|
|
# from the filelist in FAME! corpus.
|
|
|
|
#fame_functions.make_hcopy_scp_from_filelist_in_fame(default.fame_dir, dataset, feature_dir_, hcopy_scp.name)
|
|
|
|
# from the list of label files.
|
|
|
|
lab_list = glob.glob(os.path.join(label_dir_, '*.lab'))
|
|
|
|
feature_list = [
|
|
|
|
os.path.join(wav_dir_, os.path.basename(lab_file).replace('.lab', '.wav')) + '\t'
|
|
|
|
+ os.path.join(feature_dir_, os.path.basename(lab_file).replace('.lab', '.mfc'))
|
|
|
|
for lab_file in lab_list]
|
2019-03-03 02:05:37 +01:00
|
|
|
|
|
|
|
if os.path.exists(empty_mfc_file):
|
|
|
|
os.remove(empty_mfc_file)
|
2019-02-04 13:46:27 +01:00
|
|
|
with open(hcopy_scp.name, 'wb') as f:
|
|
|
|
f.write(bytes('\n'.join(feature_list), 'ascii'))
|
2018-04-25 09:07:46 +02:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
# extract features.
|
|
|
|
print(">>> extracting features on {}...".format(dataset))
|
2019-03-03 02:05:37 +01:00
|
|
|
chtk.wav2mfc(hcopy_scp.name)
|
2019-02-04 13:46:27 +01:00
|
|
|
os.remove(hcopy_scp.name)
|
2018-04-25 09:07:46 +02:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
# make hcompv.scp.
|
|
|
|
print(">>> making a script file for {}...".format(dataset))
|
|
|
|
listdir = glob.glob(os.path.join(label_dir_, '*.dic'))
|
|
|
|
mfc_list = [filename.replace(label_dir_, feature_dir_).replace('.dic', '.mfc') for filename in listdir]
|
|
|
|
hcompv_scp = os.path.join(tmp_dir, dataset + '.scp')
|
|
|
|
with open(hcompv_scp, 'wb') as f:
|
|
|
|
f.write(bytes('\n'.join(mfc_list) + '\n', 'ascii'))
|
2018-04-25 09:07:46 +02:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
print("elapsed time: {}".format(time.time() - timer_start))
|
2018-04-25 09:07:46 +02:00
|
|
|
|
|
|
|
|
|
|
|
## ======================= flat start monophones =======================
|
2019-02-04 13:46:27 +01:00
|
|
|
if flat_start:
|
|
|
|
timer_start = time.time()
|
|
|
|
print('==== flat start ====')
|
2019-03-03 02:05:37 +01:00
|
|
|
fh.make_new_directory(model0_dir, existing_dir='leave')
|
|
|
|
|
2019-03-05 00:11:38 +01:00
|
|
|
chtk.flat_start(hcompv_scp_train, model0_dir)
|
|
|
|
|
|
|
|
# create macros.
|
|
|
|
vFloors = os.path.join(model0_dir, 'vFloors')
|
|
|
|
if os.path.exists(vFloors):
|
|
|
|
chtk.create_macros(vFloors)
|
2018-04-25 09:07:46 +02:00
|
|
|
|
|
|
|
# allocate mean & variance to all phones in the phone list
|
2019-02-04 20:32:12 +01:00
|
|
|
print('>>> allocating mean & variance to all phones in the phone list...')
|
2019-03-03 02:05:37 +01:00
|
|
|
chtk.create_hmmdefs(
|
2019-02-04 20:32:12 +01:00
|
|
|
os.path.join(model0_dir, proto_name),
|
2019-03-03 02:05:37 +01:00
|
|
|
os.path.join(model0_dir, 'hmmdefs')
|
|
|
|
)
|
2019-02-04 20:32:12 +01:00
|
|
|
|
2019-02-04 13:46:27 +01:00
|
|
|
print("elapsed time: {}".format(time.time() - timer_start))
|
2018-04-25 09:07:46 +02:00
|
|
|
|
|
|
|
|
2019-02-04 20:32:12 +01:00
|
|
|
## ======================= train model without short pause =======================
|
2019-02-04 13:46:27 +01:00
|
|
|
if train_model_without_sp:
|
|
|
|
print('==== train model without sp ====')
|
2019-03-05 00:11:38 +01:00
|
|
|
|
|
|
|
timer_start = time.time()
|
|
|
|
niter = chtk.re_estimation_until_saturated(
|
|
|
|
model1_dir,
|
|
|
|
model0_dir, improvement_threshold, hcompv_scp_train,
|
|
|
|
os.path.join(htk_stimmen_dir, 'mfc'),
|
|
|
|
'mfc',
|
|
|
|
os.path.join(htk_stimmen_dir, 'word_lattice.ltc'),
|
|
|
|
mlf_file=mlf_file_train,
|
|
|
|
lexicon_file=os.path.join(htk_stimmen_dir, 'lexicon_recognition.dic')
|
|
|
|
)
|
|
|
|
|
|
|
|
print("elapsed time: {}".format(time.time() - timer_start))
|
2019-02-04 20:32:12 +01:00
|
|
|
|
|
|
|
|
|
|
|
## ======================= adding sp to the model =======================
|
|
|
|
if add_sp:
|
|
|
|
print('==== adding sp to the model ====')
|
2019-03-05 00:11:38 +01:00
|
|
|
# reference:
|
|
|
|
# http://www.f.waseda.jp/yusukekondo/htk.html#flat_start_estimation
|
2019-03-07 22:16:50 +01:00
|
|
|
timer_start = time.time()
|
2019-02-04 20:32:12 +01:00
|
|
|
|
|
|
|
# make model with sp.
|
2019-03-05 00:11:38 +01:00
|
|
|
print('>>> adding sp state to the last model in the previous step...')
|
|
|
|
fh.make_new_directory(model1sp_dir, existing_dir='leave')
|
2019-03-07 22:16:50 +01:00
|
|
|
niter = chtk.get_niter_max(model1_dir)
|
2019-03-05 00:11:38 +01:00
|
|
|
modeln_dir_pre = os.path.join(model1_dir, 'iter'+str(niter))
|
2019-03-07 22:16:50 +01:00
|
|
|
modeln_dir = os.path.join(model1sp_dir, 'iter0')
|
2019-03-05 00:11:38 +01:00
|
|
|
chtk.add_sp(modeln_dir_pre, modeln_dir)
|
2019-03-07 22:16:50 +01:00
|
|
|
print("elapsed time: {}".format(time.time() - timer_start))
|
2019-02-04 20:32:12 +01:00
|
|
|
|
2019-03-07 22:16:50 +01:00
|
|
|
niter = chtk.re_estimation_until_saturated(
|
|
|
|
model1sp_dir, modeln_dir, improvement_threshold, hcompv_scp_train,
|
|
|
|
os.path.join(htk_stimmen_dir, 'mfc'),
|
|
|
|
'mfc',
|
|
|
|
os.path.join(htk_stimmen_dir, 'word_lattice.ltc'),
|
|
|
|
mlf_file=mlf_file_train,
|
|
|
|
lexicon_file=os.path.join(htk_stimmen_dir, 'lexicon_recognition.dic'),
|
|
|
|
model_type='monophone_with_sp'
|
|
|
|
)
|
2019-02-14 00:21:28 +01:00
|
|
|
|
|
|
|
|
2019-03-07 22:16:50 +01:00
|
|
|
## ======================= train model with re-aligned mlf =======================
|
|
|
|
if train_model_with_re_aligned_mlf:
|
|
|
|
print('==== traina model with re-aligned mlf ====')
|
|
|
|
|
|
|
|
print('>>> re-aligning the training data... ')
|
|
|
|
timer_start = time.time()
|
|
|
|
niter = chtk.get_niter_max(model1sp_dir)
|
|
|
|
modeln_dir = os.path.join(model1sp_dir, 'iter'+str(niter))
|
|
|
|
chtk.make_aligned_label(
|
|
|
|
os.path.join(modeln_dir, 'macros'),
|
|
|
|
os.path.join(modeln_dir, 'hmmdefs'),
|
|
|
|
mlf_file_train_aligned,
|
|
|
|
os.path.join(label_dir, 'train_word.mlf'),
|
|
|
|
hcompv_scp_train)
|
|
|
|
print("elapsed time: {}".format(time.time() - timer_start))
|
|
|
|
|
|
|
|
print('>>> re-estimation... ')
|
|
|
|
timer_start = time.time()
|
|
|
|
fh.make_new_directory(model1sp2_dir, existing_dir='leave')
|
|
|
|
niter = chtk.get_niter_max(model1sp_dir)
|
|
|
|
niter = chtk.re_estimation_until_saturated(
|
|
|
|
model1sp2_dir,
|
|
|
|
os.path.join(model1sp_dir, 'iter'+str(niter)),
|
|
|
|
improvement_threshold,
|
|
|
|
hcompv_scp_train,
|
|
|
|
os.path.join(htk_stimmen_dir, 'mfc'),
|
|
|
|
'mfc',
|
|
|
|
os.path.join(htk_stimmen_dir, 'word_lattice.ltc'),
|
|
|
|
mlf_file=mlf_file_train,
|
|
|
|
lexicon_file=os.path.join(htk_stimmen_dir, 'lexicon_recognition.dic'),
|
|
|
|
model_type='monophone_with_sp'
|
|
|
|
)
|
|
|
|
print("elapsed time: {}".format(time.time() - timer_start))
|
2019-03-03 02:05:37 +01:00
|
|
|
|
|
|
|
|
2019-03-07 22:16:50 +01:00
|
|
|
## ======================= train triphone =======================
|
2019-03-03 02:05:37 +01:00
|
|
|
if train_triphone:
|
|
|
|
triphone_mlf = os.path.join(default.htk_dir, 'label', 'train_triphone.mlf')
|
|
|
|
macros = os.path.join(model_dir, 'hmm1_tri', 'iter0', 'macros')
|
|
|
|
hmmdefs = os.path.join(model_dir, 'hmm1_tri', 'iter0', 'hmmdefs')
|
|
|
|
model_out_dir = os.path.join(model_dir, 'hmm1_tri', 'iter1')
|
|
|
|
run_command([
|
|
|
|
'HERest', '-B',
|
|
|
|
'-C', config_train,
|
|
|
|
'-I', triphone_mlf,
|
|
|
|
'-t', '250.0', '150.0', '1000.0',
|
|
|
|
'-s', 'stats'
|
|
|
|
'-S', hcompv_scp_train,
|
|
|
|
'-H', macros,
|
|
|
|
'-H', hmmdefs,
|
|
|
|
'-M', model_out_dir,
|
|
|
|
os.path.join(config_dir, 'triphonelist.txt')
|
|
|
|
])
|