Blame view
egs/iam/v2/local/gen_topo.py
4.14 KB
8dcb6dfcb
first commit
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 |
#!/usr/bin/env python
# Copyright 2017 (author: Chun-Chieh Chang)
# Generate a topology file. This allows control of the number of states in the
# non-silence HMMs, and in the silence HMMs. This is a modified version of
# 'utils/gen_topo.pl'. The difference is that this creates two topologies for
# the non-silence HMMs. The number of states for punctuations is different than
# the number of states for other characters.
from __future__ import print_function
from __future__ import division
import argparse
import string
parser = argparse.ArgumentParser(description="Usage: steps/nnet3/chain/gen_topo.py "
"<colon-separated-nonsilence-phones> <colon-separated-silence-phones>"
"e.g.: steps/nnet3/chain/gen_topo.pl 4:5:6:7:8:9:10 1:2:3
",
epilog="See egs/swbd/s5c/local/chain/train_tdnn_a.sh for example of usage.");
parser.add_argument("num_nonsil_states", type=int, help="number of states for nonsilence phones");
parser.add_argument("num_sil_states", type=int, help="number of states for silence phones");
parser.add_argument("num_punctuation_states", type=int, help="number of states for punctuation");
parser.add_argument("nonsilence_phones",
help="List of non-silence phones as integers, separated by colons, e.g. 4:5:6:7:8:9");
parser.add_argument("silence_phones",
help="List of silence phones as integers, separated by colons, e.g. 1:2:3");
parser.add_argument("phone_list", help="file containing all phones and their corresponding number.");
args = parser.parse_args()
silence_phones = [ int(x) for x in args.silence_phones.split(":") ]
nonsilence_phones = [ int(x) for x in args.nonsilence_phones.split(":") ]
all_phones = silence_phones + nonsilence_phones
punctuation_phones = []
exclude = set("!(),.?;:'-\"")
with open(args.phone_list) as f:
for line in f:
line = line.strip()
phone = line.split(' ')[0]
if len(phone) == 1 and phone in exclude:
punctuation_phones.append(int(line.split(' ')[1]))
# For nonsilence phones that are not punctuations
print("<Topology>")
print("<TopologyEntry>")
print("<ForPhones>")
print(" ".join([str(x) for x in nonsilence_phones if x not in punctuation_phones]))
print("</ForPhones>")
for x in range(0, args.num_nonsil_states):
xp1 = x + 1
print("<State> {0} <PdfClass> {0} <Transition> {0} 0.75 <Transition> {1} 0.25 </State>".format(x, xp1))
print("<State> {} </State>".format(args.num_nonsil_states))
print("</TopologyEntry>")
# For nonsilence phones that ar punctuations
print("<TopologyEntry>")
print("<ForPhones>")
print(" ".join([str(x) for x in nonsilence_phones if x in punctuation_phones]))
print("</ForPhones>")
for x in range(0, args.num_punctuation_states):
xp1 = x + 1
print("<State> {0} <PdfClass> {0} <Transition> {0} 0.75 <Transition> {1} 0.25 </State>".format(x, xp1))
print("<State> {} </State>".format(args.num_punctuation_states))
print("</TopologyEntry>")
# For silence phones
print("<TopologyEntry>")
print("<ForPhones>")
print(" ".join([str(x) for x in silence_phones]))
print("</ForPhones>")
if(args.num_sil_states > 1):
transp = 1.0 / (args.num_sil_states - 1)
state_str = "<State> 0 <PdfClass> 0 "
for x in range(0, (args.num_sil_states - 1)):
state_str = "{}<Transition> {} {} ".format(state_str, x, transp))
state_str = state_str + "</State>"
print(state_str)
for x in range(1, (args.num_sil_states - 1)):
state_str = "<State> {0} <PdfClass {0} ".format(x))
for y in range(1, args.num_sil_states):
state_str = "{}<Transition> {} {} ".format(state_str, y, transp))
state_str = state_str + "</State>"
print(state_str)
second_last = args.num_sil_states - 1
print("<State> {0} <PdfClass> {0} <Transition> {0} 0.75 <Transition> {1} 0.25 </State>".format(second_last, args.num_sil_states))
print("<State> {} </State>".format(args.num_sil_states))
else:
print("<State> 0 <PdfClass> 0 <Transition> 0 0.75 <Transition> 1 0.25 </State>")
print("<State> {} </State>".format(args.num_sil_states))
print("</TopologyEntry>")
print("</Topology>")
|