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

Add emotion, event of SenseVoice. (#1257)

Browse Source 
* Add emotion, event of SenseVoice.

* Fix tokens size check and update java api.

https://github.com/k2-fsa/sherpa-onnx/pull/1257
This commit is contained in:
Robin Zhong
2024-08-14 15:50:13 +08:00
committed by GitHub
parent f300ec0f98
commit 62c4d4ab62
10 changed files with 95 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
sherpa-onnx/java-api/src/com/k2fsa/sherpa/onnx/OfflineRecognizer.java
View File

@@ -41,7 +41,10 @@ public class OfflineRecognizer {
String text = (String) arr[0];
String[] tokens = (String[]) arr[1];
float[] timestamps = (float[]) arr[2];
return new OfflineRecognizerResult(text, tokens, timestamps);
String lang = (String) arr[3];
String emotion = (String) arr[4];
String event = (String) arr[5];
return new OfflineRecognizerResult(text, tokens, timestamps, lang, emotion, event);
}
private native void delete(long ptr);

20
sherpa-onnx/java-api/src/com/k2fsa/sherpa/onnx/OfflineRecognizerResult.java
View File

@@ -6,11 +6,17 @@ public class OfflineRecognizerResult {
private final String text;
private final String[] tokens;
private final float[] timestamps;
private final String lang;
private final String emotion;
private final String event;
public OfflineRecognizerResult(String text, String[] tokens, float[] timestamps) {
public OfflineRecognizerResult(String text, String[] tokens, float[] timestamps, String lang, String emotion, String event) {
this.text = text;
this.tokens = tokens;
this.timestamps = timestamps;
this.lang = lang;
this.emotion = emotion;
this.event = event;
}
public String getText() {
@@ -24,4 +30,16 @@ public class OfflineRecognizerResult {
public float[] getTimestamps() {
return timestamps;
}
public String getLang() {
return lang;
}
public String getEmotion() {
return emotion;
}
public String getEvent() {
return event;
}
}