199 lines
5.8 KiB
Python
199 lines
5.8 KiB
Python
## this script should be used only by Aki Kunikoshi.
|
||
|
||
import numpy as np
|
||
import argparse
|
||
import json
|
||
|
||
from novoapi.backend import session
|
||
|
||
import os
|
||
#os.chdir(r'C:\Users\Aki\source\repos\acoustic_model\acoustic_model')
|
||
import defaultfiles as default
|
||
|
||
|
||
def load_phonset():
|
||
translation_key_ipa2novo70 = dict()
|
||
translation_key_novo702ipa = dict()
|
||
|
||
#phonelist_novo70_ = pd.ExcelFile(default.phonelist_novo70_xlsx)
|
||
#df = pd.read_excel(phonelist_novo70_, 'list')
|
||
## *_simple includes columns which has only one phone in.
|
||
#for ipa, novo70 in zip(df['IPA_simple'], df['novo70_simple']):
|
||
# if not pd.isnull(ipa):
|
||
# print('{0}:{1}'.format(ipa, novo70))
|
||
# translation_key[ipa] = novo70
|
||
#phonelist_novo70 = np.unique(list(df['novo70_simple']))
|
||
|
||
phoneset_ipa = []
|
||
phoneset_novo70 = []
|
||
with open(default.novo70_phoneset, "rt", encoding="utf-8") as fin:
|
||
lines = fin.read()
|
||
lines = lines.split('\n')
|
||
for line in lines:
|
||
words = line.split('\t')
|
||
if len(words) > 1:
|
||
novo70 = words[0]
|
||
ipa = words[1]
|
||
phoneset_ipa.append(ipa)
|
||
phoneset_novo70.append(novo70)
|
||
translation_key_ipa2novo70[ipa] = novo70
|
||
translation_key_novo702ipa[novo70] = ipa
|
||
|
||
# As per Nederlandse phoneset_aki.xlsx recieved from David
|
||
# [ɔː] oh / ohr # from ipa->novo70, only oh is used.
|
||
# [ɪː] ih / ihr # from ipa->novo70, only ih is used.
|
||
# [iː] iy
|
||
# [œː] uh
|
||
# [ɛː] eh
|
||
# [w] wv in IPA written as ʋ.
|
||
extra_ipa = ['ɔː', 'ɪː', 'iː', 'œː', 'ɛː', 'ʋ']
|
||
extra_novo70 = ['oh', 'ih', 'iy', 'uh', 'eh', 'wv']
|
||
for ipa, novo70 in zip(extra_ipa, extra_novo70):
|
||
phoneset_ipa.append(ipa)
|
||
phoneset_novo70.append(novo70)
|
||
translation_key_ipa2novo70[ipa] = novo70
|
||
translation_key_novo702ipa[novo70] = ipa
|
||
|
||
translation_key_novo702ipa['ohr'] = 'ɔː'
|
||
translation_key_novo702ipa['ihr'] = 'ɪː'
|
||
|
||
phoneset_ipa = np.unique(phoneset_ipa)
|
||
phoneset_novo70 = np.unique(phoneset_novo70)
|
||
|
||
return phoneset_ipa, phoneset_novo70, translation_key_ipa2novo70, translation_key_novo702ipa
|
||
|
||
|
||
def multi_character_tokenize(line, multi_character_tokens):
|
||
"""
|
||
Tries to match one of the tokens in multi_character_tokens at each position of line,
|
||
starting at position 0,
|
||
if so tokenizes and eats that token. Otherwise tokenizes a single character.
|
||
|
||
Copied from forced_alignment.convert_phone_set.py
|
||
"""
|
||
while line != '':
|
||
for token in multi_character_tokens:
|
||
if line.startswith(token) and len(token) > 0:
|
||
yield token
|
||
line = line[len(token):]
|
||
break
|
||
else:
|
||
yield line[:1]
|
||
line = line[1:]
|
||
|
||
|
||
def split_ipa(line):
|
||
"""
|
||
Split a line by IPA phones.
|
||
If nasalized sound (such as ɛ̃ː) is included, it will give error.
|
||
:param string line: one line written in IPA.
|
||
:return string lineSeperated: the line splitted in IPA phone.
|
||
"""
|
||
|
||
multi_character_phones = [
|
||
# IPAs in CGN.
|
||
u'ʌu', u'ɛi', u'œy', u'aː', u'eː', u'iː', u'oː', u'øː', u'ɛː', u'œː', u'ɔː', u'ɛ̃ː', u'ɑ̃ː', u'ɔ̃ː', u'œ̃', u'ɪː'
|
||
]
|
||
|
||
return [phone for phone in multi_character_tokenize(line.strip(), multi_character_phones)]
|
||
|
||
|
||
def split_novo70(line):
|
||
"""
|
||
Split a line by novo70 phones.
|
||
:param string line: one line written in novo70.
|
||
:return string lineSeperated: the line splitted by novo70 phones.
|
||
"""
|
||
_, phoneset_novo70, _, _ = load_phonset()
|
||
multi_character_phones = [p for p in phoneset_novo70 if len(p) > 1]
|
||
multi_character_phones = sorted(multi_character_phones, key=len, reverse=True)
|
||
|
||
return ['sp' if phone == ' ' else phone
|
||
for phone in multi_character_tokenize(line.strip(), multi_character_phones)]
|
||
|
||
|
||
def novo702ipa(tokens):
|
||
pronunciation = []
|
||
_, _, _, translation_key = load_phonset()
|
||
for phone in split_novo70(tokens):
|
||
pronunciation.append(translation_key.get(phone, phone))
|
||
return ' '.join(pronunciation)
|
||
|
||
|
||
# numbering of novo70 should be checked.
|
||
def ipa2novo70(tokens):
|
||
pronunciation = []
|
||
_, _, translation_key, _ = load_phonset()
|
||
for phone in split_ipa(tokens):
|
||
pronunciation.append(translation_key.get(phone, phone))
|
||
return ' '.join(pronunciation)
|
||
|
||
|
||
def make_grammar(word, pronunciation_ipa):
|
||
"""
|
||
Args:
|
||
words
|
||
pronunciation_ipa: list of pronunciation variants.
|
||
"""
|
||
#word = 'pauw'
|
||
#pronunciation_ipa = ['pau', 'pɑu']
|
||
|
||
grammer_data_elements0_pronunciation = []
|
||
for id, ipa in enumerate(pronunciation_ipa):
|
||
novo70 = ipa2novo70(ipa)
|
||
grammer_data_elements0_pronunciation.append({
|
||
"phones": novo70.split(),
|
||
"id": id
|
||
})
|
||
|
||
grammar_data = {
|
||
"kind": 'sequence',
|
||
"elements": [{
|
||
"kind": "word",
|
||
"pronunciation": grammer_data_elements0_pronunciation,
|
||
"label": word
|
||
}]
|
||
}
|
||
|
||
grammar = {
|
||
"type": "confusion_network",
|
||
"version": "1.0",
|
||
"data": grammar_data,
|
||
"return_objects": ["grammar"],
|
||
"phoneset": "novo70"
|
||
}
|
||
|
||
return grammar
|
||
|
||
|
||
def forced_alignment(wav_file, word, pronunciation_ipa):
|
||
### IMPORTANT ###
|
||
# because of this function, this script should not be uploaded / shared.
|
||
|
||
# username / password cannot be passed as artuments...
|
||
p = argparse.ArgumentParser()
|
||
p.add_argument("--user", default='martijn.wieling')
|
||
p.add_argument("--password", default='fa0Thaic')
|
||
args = p.parse_args()
|
||
|
||
rec = session.Recognizer(grammar_version="1.0", lang="nl", snodeid=101, user=args.user, password=args.password, keepopen=True) # , modeldir=modeldir)
|
||
|
||
grammar = make_grammar(word, pronunciation_ipa)
|
||
result = rec.setgrammar(grammar)
|
||
#print "Set grammar result", res
|
||
result = rec.recognize_wav(wav_file)
|
||
return result.export()
|
||
|
||
|
||
def result2pronunciation(result, word):
|
||
result_ = [result[i] for i in range(len(result)) if result[i]['label'] == word]
|
||
llh = result_[0]['llh']
|
||
phones = result_[0]['phones']
|
||
pronunciation_novo70 = [phone['label'] for phone in phones]
|
||
pronunciation_ipa = [novo702ipa(phone) for phone in pronunciation_novo70]
|
||
return pronunciation_ipa, pronunciation_novo70, llh
|
||
|
||
|
||
if __name__ == 'main':
|
||
pronunciation_ipa = ['rø:s', 'mɑn', 'mɑntsjə']
|
||
grammar = make_grammar('reus', pronunciation_ipa) |