File size: 1,913 Bytes
bd48fd1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | // Copyright (c) 2026, CIIL Anuvani-LID. Apache-2.0.
// Binding test (see tests/bindings/python_ref.py): the Java API, on a desktop JVM against the Linux build of
// libsherpa-onnx-jni.so.
import com.k2fsa.sherpa.onnx.*;
import java.io.File;
public class Main {
public static void main(String[] args) {
String model = args[0], vad = args[1], longWav = args[2];
AudioTaggingModelConfig modelConfig =
AudioTaggingModelConfig.builder().setLid(model).setNumThreads(2).setDebug(false).build();
AudioTagging tagger = new AudioTagging(
AudioTaggingConfig.builder().setModel(modelConfig).setTopK(23).build());
for (int i = 3; i < args.length; ++i) {
WaveReader reader = new WaveReader(args[i]);
OfflineStream stream = tagger.createStream();
stream.acceptWaveform(reader.getSamples(), reader.getSampleRate());
AudioEvent[] events = tagger.compute(stream);
stream.release();
StringBuilder sb = new StringBuilder();
for (int k = 0; k < events.length; ++k) {
if (k > 0) sb.append(",");
sb.append("[\"").append(events[k].getName()).append("\",").append(events[k].getIndex())
.append(",").append(events[k].getProb()).append("]");
}
System.out.println("{\"file\":\"" + new File(args[i]).getName() + "\",\"events\":[" + sb + "]}");
}
tagger.release();
LidPipeline pipeline = new LidPipeline(
LidPipelineConfig.builder().setModel(modelConfig).setVadModel(vad).build());
WaveReader reader = new WaveReader(longWav);
String json = pipeline.process(reader.getSamples(), reader.getSampleRate());
System.out.println("{\"pipeline_file\":\"" + new File(longWav).getName() + "\",\"pipeline\":" + json + "}");
pipeline.release();
}
}
|