experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #3393 from imrahul361/master

Browse Source 
Run test On Java Client
This commit is contained in:
lissyx 2020-11-05 16:30:41 +01:00 committed by GitHub
commit 12badcce1f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
native_client/java/libdeepspeech/src/androidTest/java/org/mozilla/deepspeech/libdeepspeech/test/BasicTest.java
View File

@ -32,6 +32,8 @@ public class BasicTest {
public static final String modelFile = "/data/local/tmp/test/output_graph.tflite"; public static final String modelFile = "/data/local/tmp/test/output_graph.tflite";
public static final String scorerFile = "/data/local/tmp/test/kenlm.scorer"; public static final String scorerFile = "/data/local/tmp/test/kenlm.scorer";
public static final String wavFile = "/data/local/tmp/test/LDC93S1.wav"; public static final String wavFile = "/data/local/tmp/test/LDC93S1.wav";
public static final String[] word = {"friend", "enemy", "family"};
public static final float[] boost = {1.5f, 0f, -20.4f};
private char readLEChar(RandomAccessFile f) throws IOException { private char readLEChar(RandomAccessFile f) throws IOException {
byte b1 = f.readByte(); byte b1 = f.readByte();
@ -149,4 +151,35 @@ public class BasicTest {
assertEquals("she had your dark suit in greasy wash water all year", decoded); assertEquals("she had your dark suit in greasy wash water all year", decoded);
m.freeModel(); m.freeModel();
} }
@Test
public void loadDeepSpeech_HotWord_withLM() {
DeepSpeechModel m = new DeepSpeechModel(modelFile);
m.enableExternalScorer(scorerFile);
for(int i = 0; i < word.length; i++) {
m.addHotWord(word[i], boost[i]);
String decoded = doSTT(m, false);
assertEquals("she had your dark suit in greasy wash water all year", decoded);
m.eraseHotWord(word[i]);
}
m.freeModel();
}
@Test
public void loadDeepSpeech_HotWord_noLM() {
DeepSpeechModel m = new DeepSpeechModel(modelFile);
try {
m.addHotWord(word[0], boost[0]);
assert(false);
}
catch(Exception e) {
assertEquals("Error: External scorer is not enabled. (0x2004).", e.getMessage());
}
finally {
m.freeModel();
assert(true);
}
}
} }