Add JavaScript (node-addon) API for speech enhancement GTCRN models (#1996)
This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
// Copyright (c) 2025 Xiaomi Corporation (authors: Fangjun Kuang)
|
||||
|
||||
const sherpa_onnx = require('sherpa-onnx-node');
|
||||
|
||||
function createOfflineSpeechDenoiser() {
|
||||
// please download models from
|
||||
// https://github.com/k2-fsa/sherpa-onnx/releases/tag/speech-enhancement-models
|
||||
const config = {
|
||||
model: {
|
||||
gtcrn: {model: './gtcrn_simple.onnx'},
|
||||
debug: true,
|
||||
numThreads: 1,
|
||||
},
|
||||
};
|
||||
|
||||
return new sherpa_onnx.OfflineSpeechDenoiser(config);
|
||||
}
|
||||
|
||||
const sd = createOfflineSpeechDenoiser();
|
||||
|
||||
const waveFilename = './inp_16k.wav';
|
||||
const wave = sherpa_onnx.readWave(waveFilename);
|
||||
const denoised = sd.run({
|
||||
samples: wave.samples,
|
||||
sampleRate: wave.sampleRate,
|
||||
enableExternalBuffer: true
|
||||
});
|
||||
sherpa_onnx.writeWave(
|
||||
'./enhanced-16k.wav',
|
||||
{samples: denoised.samples, sampleRate: denoised.sampleRate});
|
||||
|
||||
console.log(`Saved to ./enhanced-16k.wav`);
|
||||
Reference in New Issue
Block a user