EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Add HLG decoding for streaming CTC models (#731)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-04-03 21:31:42 +08:00
committed by GitHub
parent f8832cb5f2
commit db67e00c77
28 changed files with 668 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
sherpa-onnx/csrc/online-stream.cc
View File

@@ -104,6 +104,18 @@ class OnlineStream::Impl {
return paraformer_alpha_cache_;
}
void SetFasterDecoder(std::unique_ptr<kaldi_decoder::FasterDecoder> decoder) {
faster_decoder_ = std::move(decoder);
}
kaldi_decoder::FasterDecoder *GetFasterDecoder() const {
return faster_decoder_.get();
}
int32_t &GetFasterDecoderProcessedFrames() {
return faster_decoder_processed_frames_;
}
private:
FeatureExtractor feat_extractor_;
/// For contextual-biasing
@@ -121,6 +133,8 @@ class OnlineStream::Impl {
std::vector<float> paraformer_encoder_out_cache_;
std::vector<float> paraformer_alpha_cache_;
OnlineParaformerDecoderResult paraformer_result_;
std::unique_ptr<kaldi_decoder::FasterDecoder> faster_decoder_;
int32_t faster_decoder_processed_frames_ = 0;
};
OnlineStream::OnlineStream(const FeatureExtractorConfig &config /*= {}*/,
@@ -208,6 +222,19 @@ const ContextGraphPtr &OnlineStream::GetContextGraph() const {
return impl_->GetContextGraph();
}
void OnlineStream::SetFasterDecoder(
std::unique_ptr<kaldi_decoder::FasterDecoder> decoder) {
impl_->SetFasterDecoder(std::move(decoder));
}
kaldi_decoder::FasterDecoder *OnlineStream::GetFasterDecoder() const {
return impl_->GetFasterDecoder();
}
int32_t &OnlineStream::GetFasterDecoderProcessedFrames() {
return impl_->GetFasterDecoderProcessedFrames();
}
std::vector<float> &OnlineStream::GetParaformerFeatCache() {
return impl_->GetParaformerFeatCache();
}