| |
| |
| |
| 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(); |
| } |
| } |
|
|
