All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
merge-vads.cc File Reference
Include dependency graph for merge-vads.cc:

Go to the source code of this file.

Namespaces

 kaldi
 Relabels neural network egs with the read pdf-id alignments.
 

Functions

void PrepareMap (const std::string map_rxfilename, unordered_map< std::pair< int32, int32 >, int32, PairHasher< int32 > > *map)
 PrepareMap creates a mapping between the pairs of VAD decisions and the output label. More...
 
int main (int argc, char *argv[])
 

Function Documentation

int main ( int  argc,
char *  argv[] 
)

Definition at line 85 of file merge-vads.cc.

References ParseOptions::GetArg(), rnnlm::i, KALDI_ERR, KALDI_LOG, KALDI_WARN, ParseOptions::NumArgs(), kaldi::PrepareMap(), ParseOptions::PrintUsage(), ParseOptions::Read(), and ParseOptions::Register().

85  {
86  using namespace kaldi;
87  typedef kaldi::int32 int32;
88  try {
89  const char *usage =
90  "This program merges two archives of per-frame weights representing\n"
91  "voice activity decisions. By default, the program assumes that the\n"
92  "input vectors consist of floats that are 0.0 if a frame is judged\n"
93  "as nonspeech and 1.0 if it is considered speech. The default\n"
94  "behavior produces a frame-level decision of 1.0 if both input frames\n"
95  "are 1.0, and 0.0 otherwise. Additional classes (e.g., 2.0 for music)\n"
96  "can be handled using the \"map\" option.\n"
97  "\n"
98  "Usage: merge-vads [options] <vad-rspecifier-1> <vad-rspecifier-2>\n"
99  " <vad-wspecifier>\n"
100  "e.g.: merge-vads [options] scp:vad_energy.scp scp:vad_gmm.scp\n"
101  " ark:vad.ark\n"
102  "See also: compute-vad-from-frame-likes, compute-vad, ali-to-post,\n"
103  "post-to-weights\n";
104 
105  ParseOptions po(usage);
106  std::string map_rxfilename;
107  po.Register("map", &map_rxfilename, "This table specifies a mapping "
108  "between the labels of the frame-level decisions in the first and "
109  "second input archives to the integer output label.");
110 
111  po.Read(argc, argv);
112  if (po.NumArgs() != 3) {
113  po.PrintUsage();
114  exit(1);
115  }
116 
117  unordered_map<std::pair<int32, int32>, int32, PairHasher<int32> > map;
118  PrepareMap(map_rxfilename, &map);
119  SequentialBaseFloatVectorReader first_vad_reader(po.GetArg(1));
120  RandomAccessBaseFloatVectorReader second_vad_reader(po.GetArg(2));
121  BaseFloatVectorWriter vad_writer(po.GetArg(3));
122 
123  int32 num_done = 0, num_err = 0;
124  for (;!first_vad_reader.Done(); first_vad_reader.Next()) {
125  std::string utt = first_vad_reader.Key();
126  Vector<BaseFloat> vad1(first_vad_reader.Value());
127  if (!second_vad_reader.HasKey(utt)) {
128  KALDI_WARN << "No vector for utterance " << utt;
129  num_err++;
130  continue;
131  }
132  Vector<BaseFloat> vad2(second_vad_reader.Value(utt));
133  if (vad1.Dim() != vad2.Dim()) {
134  KALDI_WARN << "VAD length mismatch for utterance " << utt;
135  num_err++;
136  continue;
137  }
138  Vector<BaseFloat> vad_result(vad1.Dim());
139  for (int32 i = 0; i < vad1.Dim(); i++) {
140  std::pair<int32, int32> key(static_cast<int32>(vad1(i)),
141  static_cast<int32>(vad2(i)));
142  unordered_map<std::pair<int32, int32>, int32,
143  PairHasher<int32> >::const_iterator iter = map.find(key);
144  if (iter == map.end()) {
145  KALDI_ERR << "Map is missing combination "
146  << vad1(i) << " and " << vad2(i);
147  } else {
148  vad_result(i) = iter->second;
149  }
150  }
151 
152  vad_writer.Write(utt, vad_result);
153  num_done++;
154  }
155  KALDI_LOG << "Merged voice activity detection decisions; "
156  << "processed " << num_done << " utterances successfully; "
157  << num_err << " had errors.";
158  return (num_done != 0 ? 0 : 1);
159  } catch(const std::exception &e) {
160  std::cerr << e.what();
161  return -1;
162  }
163 }
Relabels neural network egs with the read pdf-id alignments.
Definition: chain.dox:20
A templated class for writing objects to an archive or script file; see The Table concept...
Definition: kaldi-table.h:366
Allows random access to a collection of objects in an archive or script file; see The Table concept...
Definition: kaldi-table.h:233
The class ParseOptions is for parsing command-line options; see Parsing command-line options for more...
Definition: parse-options.h:36
A templated class for reading objects sequentially from an archive or script file; see The Table conc...
Definition: kaldi-table.h:287
void PrepareMap(const std::string &map_rxfilename, int32 num_classes, unordered_map< int32, int32 > *map)
PrepareMap creates a map that specifies the mapping between the input and output class labels...
#define KALDI_ERR
Definition: kaldi-error.h:127
#define KALDI_WARN
Definition: kaldi-error.h:130
#define KALDI_LOG
Definition: kaldi-error.h:133
A hashing function-object for pairs of ints.
Definition: stl-utils.h:237