[src] Add GetSamplingFrequency() to class FeaturePipelineInfo (#3939)

kaldi-asr · Mar 3, 2020 · beb1c4b · beb1c4b
1 parent 194f674
commit beb1c4b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/online2/online-nnet2-feature-pipeline.cc b/src/online2/online-nnet2-feature-pipeline.cc
@@ -241,5 +241,17 @@ BaseFloat OnlineNnet2FeaturePipelineInfo::FrameShiftInSeconds() const {
   }
 }
 
+BaseFloat OnlineNnet2FeaturePipelineInfo::GetSamplingFrequency() {
+  if (feature_type == "mfcc") {
+    return mfcc_opts.frame_opts.samp_freq;
+  } else if (feature_type == "plp") {
+    return plp_opts.frame_opts.samp_freq;
+  } else if (feature_type == "fbank") {
+    return fbank_opts.frame_opts.samp_freq;
+  } else {
+    KALDI_ERR << "Unknown feature type " << feature_type;
+  }
+  return 0.0f; // avoiding a possible "return missing" warning
+}
 
 }  // namespace kaldi
diff --git a/src/online2/online-nnet2-feature-pipeline.h b/src/online2/online-nnet2-feature-pipeline.h
@@ -176,6 +176,9 @@ struct OnlineNnet2FeaturePipelineInfo {
   /// on the command line instead of inside sub-config-files.
   OnlineSilenceWeightingConfig silence_weighting_config;
 
+  /// Returns the frequency expected by the model
+  BaseFloat GetSamplingFrequency();
+
   int32 IvectorDim() { return ivector_extractor_info.extractor.IvectorDim(); }
  private:
   KALDI_DISALLOW_COPY_AND_ASSIGN(OnlineNnet2FeaturePipelineInfo);