33 lines
870 B
JavaScript
33 lines
870 B
JavaScript
// 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`);
|