Add Java API for speech enhancement GTCRN models (#2009)

This commit is contained in:
Fangjun Kuang
2025-03-16 15:13:20 +08:00
committed by GitHub
parent ed8e6c9aed
commit 71824992a7
10 changed files with 294 additions and 0 deletions

View File

@@ -0,0 +1,43 @@
// Copyright 2025 Xiaomi Corporation
package com.k2fsa.sherpa.onnx;
public class OfflineSpeechDenoiser {
static {
System.loadLibrary("sherpa-onnx-jni");
}
private long ptr = 0;
public OfflineSpeechDenoiser(OfflineSpeechDenoiserConfig config) {
ptr = newFromFile(config);
}
public int getSampleRate() {
return getSampleRate(ptr);
}
public DenoisedAudio run(float[] samples, int sampleRate) {
return run(ptr, samples, sampleRate);
}
protected void finalize() throws Throwable {
release();
}
public void release() {
if (this.ptr == 0) {
return;
}
delete(this.ptr);
this.ptr = 0;
}
private native void delete(long ptr);
private native int getSampleRate(long ptr);
private native DenoisedAudio run(long ptr, float[] samples, int sampleRate);
private native long newFromFile(OfflineSpeechDenoiserConfig config);
}