izaya/LinVAM
1
0
Fork 0
mirror of https://github.com/aidygus/LinVAM.git synced 2024-11-23 09:18:04 +11:00
Code Issues Releases Wiki Activity

Update another Decoder config, remove whitespaces

Browse Source
This commit is contained in:
smirgol 2023-06-29 09:28:40 +02:00
parent 32d30656c4
commit 407b3e3d04
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
set_kws_threshold.py
View File

@ -15,7 +15,7 @@ from sphinxbase.sphinxbase import *
# keyphrases found in kwlist # keyphrases found in kwlist
WORDS = [] WORDS = []
# test case containing multiple occurances # test case containing multiple occurances
# of words to be used as training audio # of words to be used as training audio
TEST_CASE = [] TEST_CASE = []
# Threshold values # Threshold values
@ -36,7 +36,7 @@ def preprocess_files(dic_path, kwlist_path):
with open(dic_path) as _f: with open(dic_path) as _f:
_content = _f.readlines() _content = _f.readlines()
_content = [x.strip() for x in _content] _content = [x.strip() for x in _content]
with open(kwlist_path) as _f: with open(kwlist_path) as _f:
WORDS = _f.readlines() WORDS = _f.readlines()
WORDS = [x.strip()[:x.strip().rfind(' ')] for x in WORDS] WORDS = [x.strip()[:x.strip().rfind(' ')] for x in WORDS]
@ -188,7 +188,7 @@ def actual_tuning(dic_path, kwlist_path, _z):
_previous_missed.extend(_missed) _previous_missed.extend(_missed)
_previous_fa = [] _previous_fa = []
_previous_fa.extend(_fa) _previous_fa.extend(_fa)
_missed, _fa = process_threshold(kws_analysis(dic_path, kwlist_path)) _missed, _fa = process_threshold(kws_analysis(dic_path, kwlist_path))
if _z == 1: if _z == 1:
@ -224,14 +224,14 @@ def kws_analysis(dic, kwlist):
modeldir = "/usr/local/share/pocketsphinx/model/" modeldir = "/usr/local/share/pocketsphinx/model/"
# Create a decoder with certain model # Create a decoder with certain model
config = Decoder.default_config() config = Config(
config.set_string('-hmm', os.path.join(modeldir, 'en-us/en-us')) hmm=os.path.join(modeldir, 'en-us/en-us'),
config.set_string('-dict', dic) dict=dic,
config.set_string('-kws', kwlist) kws=kwlist,
config.set_string('-dither', "no") dither="no",
config.set_string('-logfn', '/dev/null') featparams=os.path.join(os.path.join(modeldir, 'en-us/en-us'), "feat.params"),
config.set_string('-featparams', os.path.join(os.path.join(modeldir, logfn='/dev/null'
'en-us/en-us'), "feat.params")) )
stream = open(OUTPUT_FILENAME, "rb") stream = open(OUTPUT_FILENAME, "rb")
@ -291,7 +291,7 @@ def process_threshold(analysis_result):
missed[position_original][1] += 1 missed[position_original][1] += 1
print ('Missed ', val) print ('Missed ', val)
return missed, false_alarms return missed, false_alarms
if __name__ == '__main__': if __name__ == '__main__':
DIC_FILE = "/home/pankaj/catkin_ws/src/pocketsphinx/demo/voice_cmd.dic" DIC_FILE = "/home/pankaj/catkin_ws/src/pocketsphinx/demo/voice_cmd.dic"
KWLIST_FILE = "/home/pankaj/catkin_ws/src/pocketsphinx/demo/automated.kwlist" KWLIST_FILE = "/home/pankaj/catkin_ws/src/pocketsphinx/demo/automated.kwlist"