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

Implemented tokens and timestamps in Python API (#205)

Browse Source
This commit is contained in:
Wilson Wongso
2023-07-12 08:12:31 +07:00
committed by GitHub
parent 0b5fa24134
commit b2364b0374
2 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
sherpa-onnx/python/csrc/online-recognizer.cc
View File

@@ -14,7 +14,12 @@ namespace sherpa_onnx {
static void PybindOnlineRecognizerResult(py::module *m) { static void PybindOnlineRecognizerResult(py::module *m) {
using PyClass = OnlineRecognizerResult; using PyClass = OnlineRecognizerResult;
py::class_<PyClass>(*m, "OnlineRecognizerResult") py::class_<PyClass>(*m, "OnlineRecognizerResult")
.def_property_readonly("text", [](PyClass &self) { return self.text; }); .def_property_readonly(
"text", [](PyClass &self) -> std::string { return self.text; })
.def_property_readonly(
"tokens", [](PyClass &self) -> std::vector<std::string> { return self.tokens; })
.def_property_readonly(
"timestamps", [](PyClass &self) -> std::vector<float> { return self.timestamps; });
} }
static void PybindOnlineRecognizerConfig(py::module *m) { static void PybindOnlineRecognizerConfig(py::module *m) {

6
sherpa-onnx/python/sherpa_onnx/online_recognizer.py
View File

@@ -149,6 +149,12 @@ class OnlineRecognizer(object):
def get_result(self, s: OnlineStream) -> str: def get_result(self, s: OnlineStream) -> str:
return self.recognizer.get_result(s).text.strip() return self.recognizer.get_result(s).text.strip()
def tokens(self, s: OnlineStream) -> List[str]:
return self.recognizer.get_result(s).tokens
def timestamps(self, s: OnlineStream) -> List[float]:
return self.recognizer.get_result(s).timestamps
def is_endpoint(self, s: OnlineStream) -> bool: def is_endpoint(self, s: OnlineStream) -> bool:
return self.recognizer.is_endpoint(s) return self.recognizer.is_endpoint(s)