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

OfflineRecognizer supports create stream with hotwords (#1833)

Browse Source 
Co-authored-by: Wangkai <kell.wang@huawei.com>
This commit is contained in:
Kell
2025-02-10 16:26:56 +08:00
committed by GitHub
parent 9559a10bd3
commit 2ac41d3d85
4 changed files with 26 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
sherpa-onnx/c-api/c-api.cc
View File

@@ -533,6 +533,13 @@ const SherpaOnnxOfflineStream *SherpaOnnxCreateOfflineStream(
return stream;
}
const SherpaOnnxOfflineStream *SherpaOnnxCreateOfflineStreamWithHotwords(
const SherpaOnnxOfflineRecognizer *recognizer, const char *hotwords) {
SherpaOnnxOfflineStream *stream =
new SherpaOnnxOfflineStream(recognizer->impl->CreateStream(hotwords));
return stream;
}
void SherpaOnnxDestroyOfflineStream(const SherpaOnnxOfflineStream *stream) {
delete stream;
}

10
sherpa-onnx/c-api/c-api.h
View File

@@ -484,6 +484,16 @@ SHERPA_ONNX_API void SherpaOnnxDestroyOfflineRecognizer(
SHERPA_ONNX_API const SherpaOnnxOfflineStream *SherpaOnnxCreateOfflineStream(
const SherpaOnnxOfflineRecognizer *recognizer);
/// Create an offline stream for accepting wave samples with the specified hot
/// words.
///
/// @param recognizer A pointer returned by SherpaOnnxCreateOfflineRecognizer()
/// @return Return a pointer to an OfflineStream. The user has to invoke
/// SherpaOnnxDestroyOfflineStream() to free it to avoid memory leak.
SHERPA_ONNX_API const SherpaOnnxOfflineStream *
SherpaOnnxCreateOfflineStreamWithHotwords(
const SherpaOnnxOfflineRecognizer *recognizer, const char *hotwords);
/// Destroy an offline stream.
///
/// @param stream A pointer returned by SherpaOnnxCreateOfflineStream()

9
sherpa-onnx/c-api/cxx-api.cc
View File

@@ -266,8 +266,13 @@ void OfflineRecognizer::Destroy(const SherpaOnnxOfflineRecognizer *p) const {
}
OfflineStream OfflineRecognizer::CreateStream() const {
auto p = SherpaOnnxCreateOfflineStream(p_);
return OfflineStream{p};
auto s = SherpaOnnxCreateOfflineStream(p_);
return OfflineStream{s};
}
OfflineStream OfflineRecognizer::CreateStream(const std::string &hotwords) const {
auto s = SherpaOnnxCreateOfflineStreamWithHotwords(p_, hotwords.c_str());
return OfflineStream{s};
}
void OfflineRecognizer::Decode(const OfflineStream *s) const {

2
sherpa-onnx/c-api/cxx-api.h
View File

@@ -301,6 +301,8 @@ class SHERPA_ONNX_API OfflineRecognizer
OfflineStream CreateStream() const;
OfflineStream CreateStream(const std::string &hotwords) const;
void Decode(const OfflineStream *s) const;
void Decode(const OfflineStream *ss, int32_t n) const;