ali-to-phones.cc
5.01 KB
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
// bin/ali-to-phones.cc
// Copyright 2009-2011 Microsoft Corporation
// 2015 IMSL, PKU-HKUST (author: Wei Shi)
// See ../../COPYING for clarification regarding multiple authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED
// WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE,
// MERCHANTABLITY OR NON-INFRINGEMENT.
// See the Apache 2 License for the specific language governing permissions and
// limitations under the License.
#include "base/kaldi-common.h"
#include "hmm/transition-model.h"
#include "hmm/hmm-utils.h"
#include "util/common-utils.h"
#include "fst/fstlib.h"
int main(int argc, char *argv[]) {
using namespace kaldi;
typedef kaldi::int32 int32;
try {
const char *usage =
"Convert model-level alignments to phone-sequences (in integer, "
"not text, form)\n"
"Usage: ali-to-phones [options] <model> <alignments-rspecifier> "
"<phone-transcript-wspecifier|ctm-wxfilename>\n"
"e.g.: \n"
" ali-to-phones 1.mdl ark:1.ali ark:-\n"
"or:\n"
" ali-to-phones --ctm-output 1.mdl ark:1.ali 1.ctm\n"
"See also: show-alignments lattice-align-phones, compare-int-vector\n";
ParseOptions po(usage);
bool per_frame = false;
bool write_lengths = false;
bool ctm_output = false;
BaseFloat frame_shift = 0.01;
po.Register("ctm-output", &ctm_output,
"If true, output the alignments in ctm format "
"(the confidences will be set to 1)");
po.Register("frame-shift", &frame_shift,
"frame shift used to control the times of the ctm output");
po.Register("per-frame", &per_frame,
"If true, write out the frame-level phone alignment "
"(else phone sequence)");
po.Register("write-lengths", &write_lengths,
"If true, write the #frames for each phone (different format)");
po.Read(argc, argv);
KALDI_ASSERT(!(per_frame && write_lengths) && "Incompatible options.");
if (po.NumArgs() != 3) {
po.PrintUsage();
exit(1);
}
std::string model_filename = po.GetArg(1),
alignments_rspecifier = po.GetArg(2);
TransitionModel trans_model;
ReadKaldiObject(model_filename, &trans_model);
SequentialInt32VectorReader reader(alignments_rspecifier);
std::string empty;
Int32VectorWriter phones_writer(ctm_output ? empty :
(write_lengths ? empty : po.GetArg(3)));
Int32PairVectorWriter pair_writer(ctm_output ? empty :
(write_lengths ? po.GetArg(3) : empty));
std::string ctm_wxfilename(ctm_output ? po.GetArg(3) : empty);
Output ctm_writer(ctm_wxfilename, false);
if (ctm_output) {
ctm_writer.Stream() << std::fixed;
ctm_writer.Stream().precision(frame_shift >= 0.01 ? 2 : 3);
}
int32 n_done = 0;
for (; !reader.Done(); reader.Next()) {
std::string key = reader.Key();
const std::vector<int32> &alignment = reader.Value();
std::vector<std::vector<int32> > split;
SplitToPhones(trans_model, alignment, &split);
if (ctm_output) {
BaseFloat phone_start = 0.0;
for (size_t i = 0; i < split.size(); i++) {
KALDI_ASSERT(!split[i].empty());
int32 phone = trans_model.TransitionIdToPhone(split[i][0]);
int32 num_repeats = split[i].size();
ctm_writer.Stream() << key << " 1 " << phone_start << " "
<< (frame_shift * num_repeats) << " " << phone << std::endl;
phone_start += frame_shift * num_repeats;
}
} else if (!write_lengths) {
std::vector<int32> phones;
for (size_t i = 0; i < split.size(); i++) {
KALDI_ASSERT(!split[i].empty());
int32 phone = trans_model.TransitionIdToPhone(split[i][0]);
int32 num_repeats = split[i].size();
//KALDI_ASSERT(num_repeats!=0);
if (per_frame)
for(int32 j = 0; j < num_repeats; j++)
phones.push_back(phone);
else
phones.push_back(phone);
}
phones_writer.Write(key, phones);
} else {
std::vector<std::pair<int32, int32> > pairs;
for (size_t i = 0; i < split.size(); i++) {
KALDI_ASSERT(split[i].size() > 0);
int32 phone = trans_model.TransitionIdToPhone(split[i][0]);
int32 num_repeats = split[i].size();
//KALDI_ASSERT(num_repeats!=0);
pairs.push_back(std::make_pair(phone, num_repeats));
}
pair_writer.Write(key, pairs);
}
n_done++;
}
KALDI_LOG << "Done " << n_done << " utterances.";
} catch(const std::exception &e) {
std::cerr << e.what();
return -1;
}
}