Add HLG decoding for streaming CTC models (#731)

2024-04-03 21:31:42 +08:00
parent f8832cb5f2
commit db67e00c77
28 changed files with 668 additions and 82 deletions
--- a/sherpa-onnx/csrc/online-ctc-greedy-search-decoder.h
+++ b/sherpa-onnx/csrc/online-ctc-greedy-search-decoder.h
@@ -17,7 +17,8 @@ class OnlineCtcGreedySearchDecoder : public OnlineCtcDecoder {
      : blank_id_(blank_id) {}

  void Decode(Ort::Value log_probs,
-              std::vector<OnlineCtcDecoderResult> *results) override;
+              std::vector<OnlineCtcDecoderResult> *results,
+              OnlineStream **ss = nullptr, int32_t n = 0) override;

 private:
  int32_t blank_id_;